File tree Expand file tree Collapse file tree 1 file changed +4
-7
lines changed
Expand file tree Collapse file tree 1 file changed +4
-7
lines changed Original file line number Diff line number Diff line change 6262 is_basic_idx ,
6363)
6464from pytensor .tensor .type import TensorType
65- from pytensor .tensor .type_other import NoneConst , NoneTypeT , SliceConstant , SliceType
65+ from pytensor .tensor .type_other import NoneConst , NoneTypeT , SliceType
6666from pytensor .tensor .variable import TensorVariable
6767
6868from pymc .logprob .abstract import (
@@ -290,14 +290,11 @@ def find_measurable_index_mixture(fgraph, node):
290290 # but the Mixture logprob assumes all mixture values are independent
291291 if any (
292292 (
293- isinstance (indices , (type (NoneConst ) | type (None )))
294- or (
295- indices .dtype .startswith ("int" )
296- and sum (1 - b for b in indices .type .broadcastable ) > 0
297- )
293+ isinstance (indices , TensorVariable )
294+ and indices .dtype .startswith ("int" )
295+ and any (not b for b in indices .type .broadcastable )
298296 )
299297 for indices in mixing_indices
300- if not isinstance (indices , SliceConstant )
301298 ):
302299 return None
303300
You can’t perform that action at this time.
0 commit comments