From a2ef7a7361a3fc3fd55cebaeb8271faf8e75a9a9 Mon Sep 17 00:00:00 2001 From: Dhruvanshu-Joshi Date: Tue, 21 Nov 2023 16:29:03 +0530 Subject: [PATCH] resolving changes --- tests/logprob/test_transforms.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/logprob/test_transforms.py b/tests/logprob/test_transforms.py index 5c88a8ff0e0..1db159d2520 100644 --- a/tests/logprob/test_transforms.py +++ b/tests/logprob/test_transforms.py @@ -688,7 +688,8 @@ def test_discrete_rv_unary_transform_fails(): conditional_logp({y_rv: y_rv.clone()}) -def test_discrete_rv_multinary_transform_fails(): +# add 2 tests. One fir supported and one for unsupported +def test_discrete_rv_multinary_transform(): y_rv = 5 + pt.random.poisson(1) with pytest.raises(RuntimeError, match="could not be derived"): conditional_logp({y_rv: y_rv.clone()})