Fix test incompatibility with newer PyTensor check

pymc-devs · Dec 5, 2024 · 50e034b · 50e034b
1 parent 2fcea58
commit 50e034b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/distributions/test_custom.py b/tests/distributions/test_custom.py
@@ -287,7 +287,7 @@ def test_multivariate_insufficient_signature(self):
         with pytest.raises(
             NotImplementedError, match="signature is not sufficient to infer the support shape"
         ):
-            CustomDist.dist(signature="(n)->(m)")
+            CustomDist.dist([0], signature="(n)->(m)")
 
 
 class TestCustomSymbolicDist: