Fix dummy

pyro-ppl · Jan 12, 2024 · c7910a4 · c7910a4
1 parent ae105d4
commit c7910a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/contrib/test_zuko.py b/tests/contrib/test_zuko.py
@@ -28,7 +28,7 @@ def test_ZukoToPyro(multivariate: bool, rsample_and_log_prob: bool):
 
     if rsample_and_log_prob:
         def dummy(self, shape):
-            x = self.rsample(x)
+            x = self.rsample(shape)
             return x, self.log_prob(x)
 
         dist.rsample_and_log_prob = dummy