Used to_tuple for size

Author: Sayam753

pymc3/distributions/simulator.py

@@ -18,7 +18,7 @@
 
 from scipy.spatial import cKDTree
 
-from pymc3.distributions.distribution import NoDistribution, draw_values
+from pymc3.distributions.distribution import NoDistribution, draw_values, to_tuple
 
 __all__ = ["Simulator"]
 
@@ -114,8 +114,9 @@ def random(self, point=None, size=None):
         -------
         array
         """
+        size = to_tuple(size)
         params = draw_values([*self.params], point=point, size=size)
-        if (size is None) or (len(size) == 0):
+        if len(size) == 0:
             return self.function(*params)
         else:
             return np.array([self.function(*params) for _ in range(size[0])])