From a714b2485769b358195e1a1fd5bb69a337fcf526 Mon Sep 17 00:00:00 2001 From: Ricardo Vieira Date: Thu, 5 Dec 2024 17:18:02 +0100 Subject: [PATCH] Fix test incompatibility with newer PyTensor check --- tests/distributions/test_custom.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/distributions/test_custom.py b/tests/distributions/test_custom.py index 5b1de161789..5f2cef54f02 100644 --- 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: