From c7bb63a65e0540c82d3ec28f7ed8468ea428e27e Mon Sep 17 00:00:00 2001 From: Sichao He <1310722434@qq.com> Date: Sun, 15 Dec 2024 22:55:34 +0800 Subject: [PATCH] Update test_ndarray.py --- brainpy/_src/math/tests/test_ndarray.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/brainpy/_src/math/tests/test_ndarray.py b/brainpy/_src/math/tests/test_ndarray.py index a09129129..e9acff357 100644 --- a/brainpy/_src/math/tests/test_ndarray.py +++ b/brainpy/_src/math/tests/test_ndarray.py @@ -62,7 +62,7 @@ def _f(self, b): def test_tracing(self): print(self.f(1.)) - with self.assertRaises(RuntimeError): + with self.assertRaises(jax.errors.UnexpectedTracerError): print(self.f(bm.ones(10)))