- Replace tt.max and tt.min with tt.switch

pymc-devs · Dec 21, 2020 · 9f9df0e · 9f9df0e
1 parent 64009ec
commit 9f9df0e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pymc3/distributions/discrete.py b/pymc3/distributions/discrete.py
@@ -930,8 +930,9 @@ def logp(self, value):
             - betaln(n - value + 1, bad - n + value + 1)
             - betaln(tot + 1, 1)
         )
-        lower = tt.max([0, n - N + k])
-        upper = tt.min([k, n])
+        # value in [max(0, n - N + k), min(k, n)]
+        lower = tt.switch(tt.gt(n - N + k, 0), n - N + k, 0)
+        upper = tt.switch(tt.lt(k, n), k, n)
         return bound(result, lower <= value, value <= upper)