union 2 random variables

Maxim Kochurov · Maxim Kochurov · commit 0c9ae26ff846 · 2022-11-16T09:01:03.000+03:00
diff --git a/pymc/distributions/logprob.py b/pymc/distributions/logprob.py
@@ -116,8 +116,7 @@ def _check_no_rvs(logp_terms: Sequence[TensorVariable]):
         if (
             node.owner
             and isinstance(node.owner.op, RandomVariable)
-            and not isinstance(node.owner.op, SimulatorRV)
-            and not isinstance(node.owner.op, MinibatchIndexRV)
+            and not isinstance(node.owner.op, (SimulatorRV, MinibatchIndexRV))
         )
     ]
     if unexpected_rv_nodes:

Original file line number	Diff line number	Diff line change
`@@ -116,8 +116,7 @@ def _check_no_rvs(logp_terms: Sequence[TensorVariable]):`
`116`	`116`	`if (`
`117`	`117`	`node.owner`
`118`	`118`	`and isinstance(node.owner.op, RandomVariable)`
`119`		`- and not isinstance(node.owner.op, SimulatorRV)`
`120`		`- and not isinstance(node.owner.op, MinibatchIndexRV)`
	`119`	`+ and not isinstance(node.owner.op, (SimulatorRV, MinibatchIndexRV))`
`121`	`120`	`)`
`122`	`121`	`]`
`123`	`122`	`if unexpected_rv_nodes:`