Reducing redundant code in discrete order

pymc-devs · Oct 23, 2023 · 894ff27 · 894ff27
1 parent 9fddde6
commit 894ff27
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/pymc/logprob/order.py b/pymc/logprob/order.py
@@ -108,10 +108,7 @@ def find_measurable_max(fgraph: FunctionGraph, node: Node) -> Optional[List[Tens
 
     # distinguish measurable discrete and continuous (because logprob is different)
     if base_var.owner.op.dtype.startswith("int"):
-        if isinstance(base_var.owner.op, RandomVariable):
-            measurable_max = MeasurableMaxDiscrete(list(axis))
-        else:
-            return None
+        measurable_max = MeasurableMaxDiscrete(list(axis))
     else:
         measurable_max = MeasurableMax(list(axis))