Skip to content

Commit 894ff27

Browse files
Reducing redundant code in discrete order
1 parent 9fddde6 commit 894ff27

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

pymc/logprob/order.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -108,10 +108,7 @@ def find_measurable_max(fgraph: FunctionGraph, node: Node) -> Optional[List[Tens
108108

109109
# distinguish measurable discrete and continuous (because logprob is different)
110110
if base_var.owner.op.dtype.startswith("int"):
111-
if isinstance(base_var.owner.op, RandomVariable):
112-
measurable_max = MeasurableMaxDiscrete(list(axis))
113-
else:
114-
return None
111+
measurable_max = MeasurableMaxDiscrete(list(axis))
115112
else:
116113
measurable_max = MeasurableMax(list(axis))
117114

0 commit comments

Comments
 (0)