Grouped types into theano_constant tuple

Author: lucianopaz

pymc3/distributions/distribution.py

@@ -9,7 +9,7 @@
     Model, get_named_nodes_and_relations, FreeRV,
     ObservedRV, MultiObservedRV, Context, InitContextMeta
 )
-from ..vartypes import string_types
+from ..vartypes import string_types, theano_constant
 from .shape_utils import (
     to_tuple,
     get_broadcastable_dist_samples,
@@ -92,7 +92,7 @@ def getattr_value(self, val):
         if isinstance(val, tt.TensorVariable):
             return val.tag.test_value
 
-        if isinstance(val, (tt.TensorConstant, theano.gof.graph.Constant)):
+        if isinstance(val, theano_constant):
             return val.value
 
         return val
@@ -502,8 +502,7 @@ def __init__(self):
 def is_fast_drawable(var):
     return isinstance(var, (numbers.Number,
                             np.ndarray,
-                            tt.TensorConstant,
-                            theano.gof.graph.Constant,
+                            theano_constant,
                             tt.sharedvar.SharedVariable))
 
 
@@ -593,8 +592,7 @@ def draw_values(params, point=None, size=None):
             if (next_, size) in drawn:
                 # If the node already has a givens value, skip it
                 continue
-            elif isinstance(next_, (tt.TensorConstant,
-                                    theano.gof.graph.Constant,
+            elif isinstance(next_, (theano_constant,
                                     tt.sharedvar.SharedVariable)):
                 # If the node is a theano.tensor.TensorConstant or a
                 # theano.tensor.sharedvar.SharedVariable, its value will be
@@ -785,7 +783,7 @@ def _draw_value(param, point=None, givens=None, size=None):
     """
     if isinstance(param, (numbers.Number, np.ndarray)):
         return param
-    elif isinstance(param, (tt.TensorConstant, theano.gof.graph.Constant)):
+    elif isinstance(param, theano_constant):
         return param.value
     elif isinstance(param, tt.sharedvar.SharedVariable):
         return param.get_value()

pymc3/vartypes.py

@@ -1,20 +1,26 @@
-__all__ = ['bool_types', 'int_types', 'float_types', 'complex_types', 'continuous_types',
-           'discrete_types', 'typefilter', 'isgenerator']
-
-bool_types = set(['int8'])
-
-int_types = set(['int8',
-                 'int16',
-                 'int32',
-                 'int64',
-                 'uint8',
-                 'uint16',
-                 'uint32',
-                 'uint64'])
-float_types = set(['float32',
-                   'float64'])
-complex_types = set(['complex64',
-                     'complex128'])
+from theano.tensor import Constant as tensor_constant
+from theano.gof.graph import Constant as graph_constant
+
+
+__all__ = [
+    "bool_types",
+    "int_types",
+    "float_types",
+    "complex_types",
+    "continuous_types",
+    "discrete_types",
+    "typefilter",
+    "isgenerator",
+    "theano_constant",
+]
+
+bool_types = set(["int8"])
+
+int_types = set(
+    ["int8", "int16", "int32", "int64", "uint8", "uint16", "uint32", "uint64"]
+)
+float_types = set(["float32", "float64"])
+complex_types = set(["complex64", "complex128"])
 continuous_types = float_types | complex_types
 discrete_types = bool_types | int_types
 
@@ -27,4 +33,7 @@ def typefilter(vars, types):
 
 
 def isgenerator(obj):
-    return hasattr(obj, '__next__')
+    return hasattr(obj, "__next__")
+
+
+theano_constant = (tensor_constant, graph_constant)