moves string_types to vartypes module

Author: tyarkoni

pymc3/distributions/distribution.py

@@ -4,18 +4,12 @@
 
 from ..memoize import memoize
 from ..model import Model, get_named_nodes
-import sys
+from ..vartypes import string_types
 
 
 __all__ = ['DensityDist', 'Distribution', 'Continuous', 'Discrete', 'NoDistribution', 'TensorType', 'draw_values']
 
 
-if sys.version_info[0] == 3:
-    string_types = str
-else:
-    string_types = basestring
-
-
 class Distribution(object):
     """Statistical distribution"""
     def __new__(cls, name, *args, **kwargs):

pymc3/vartypes.py

@@ -1,3 +1,5 @@
+import sys
+
 __all__ = ['bool_types', 'int_types', 'float_types', 'complex_types', 'continuous_types',
         'discrete_types', 'default_type', 'typefilter']
 
@@ -21,6 +23,11 @@
 default_type = {'discrete': 'int64',
                 'continuous': 'float64'}
 
+if sys.version_info[0] == 3:
+    string_types = str
+else:
+    string_types = basestring
+
 
 def typefilter(vars, types):
     # Returns variables of type `types` from `vars`