Fix test

OpenSourceEconomics · Dec 11, 2023 · abecb7f · abecb7f
1 parent 891151d
commit abecb7f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_functools.py b/tests/test_functools.py
@@ -218,7 +218,7 @@ def f(a, *, b):
         # TypeError since b is keyword-only
         f_vmapped(a, b)
 
-    with pytest.raises(ValueError, match="vmap in_axes specification"):
+    with pytest.raises(ValueError, match="vmap in_axes must be an int"):
         # ValueError since vmap doesn't support keyword arguments
         f_vmapped(a, b=b)