Update test_ndarray.py

brainpy · Dec 15, 2024 · c7bb63a · c7bb63a
1 parent c289edd
commit c7bb63a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)))