test

pymc3/distributions/dist_math.py

@@ -10,15 +10,15 @@
 
 from .special import gammaln, multigammaln
 
-def bound(logp, *conditions, broadcast_conditions=True):
+def bound(logp, *conditions, **kwargs):
     """
     Bounds a log probability density with several conditions.
 
     Parameters
     ----------
     logp : float
     *conditions : booleans
-    broadcast_conditions : bool (optional)
+    broadcast_conditions : bool (optional, default=True)
         If True, broadcasts logp to match the largest shape of the conditions.
         This is used e.g. in DiscreteUniform where logp is a scalar constant and the shape
         is specified via the conditions.
@@ -29,6 +29,8 @@ def bound(logp, *conditions, broadcast_conditions=True):
     -------
     logp with elements set to -inf where any condition is False
     """
+    broadcast_conditions = kwargs.get('broadcast_conditions', True)
+
     if broadcast_conditions:
         alltrue = alltrue_elemwise
     else: