Add flag to disable bounds check.

Author: twiecki

pymc3/distributions/dist_math.py

@@ -33,6 +33,7 @@
 
 from pymc3.distributions.shape_utils import to_tuple
 from pymc3.distributions.special import gammaln
+from pymc3.model import modelcontext
 from pymc3.theanof import floatX
 
 f = floatX
@@ -67,6 +68,13 @@ def bound(logp, *conditions, **kwargs):
     -------
     logp with elements set to -inf where any condition is False
     """
+
+    # If called inside a model context, see if bounds check is disabled
+    model = modelcontext(kwargs.get("model"))
+    if model is not None:
+        if model.disable_bounds_check:
+            return logp
+
     broadcast_conditions = kwargs.get("broadcast_conditions", True)
 
     if broadcast_conditions:

pymc3/model.py

@@ -809,6 +809,11 @@ class Model(Factor, WithMemoization, metaclass=ContextMeta):
         temporarily in the model context. See the documentation
         of theano for a complete list. Set config key
         ``compute_test_value`` to `raise` if it is None.
+    disable_bounds_check: bool
+        Disable checks that ensure that input parameters to distributions
+        are in a valid range. If your model is built in a way where you
+        know your parameters can only take on valid values you can disable
+        this for increased speed.
 
     Examples
     --------
@@ -895,11 +900,14 @@ def __new__(cls, *args, **kwargs):
         instance._theano_config = theano_config
         return instance
 
-    def __init__(self, name="", model=None, theano_config=None, coords=None):
+    def __init__(
+        self, name="", model=None, theano_config=None, coords=None, disable_bounds_check=False
+    ):
         self.name = name
         self.coords = {}
         self.RV_dims = {}
         self.add_coords(coords)
+        self.disable_bounds_check = disable_bounds_check
 
         if self.parent is not None:
             self.named_vars = treedict(parent=self.parent.named_vars)