Fix to sampler.

Fix to issue pymc-devs#3481, from @lucianopaz.
rpgoldman · May 20, 2019 · cc9a46c · cc9a46c
1 parent ff7f047
commit cc9a46c
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pymc3/distributions/continuous.py b/pymc3/distributions/continuous.py
@@ -664,8 +664,13 @@ def random(self, point=None, size=None):
         -------
         array
         """
-        mu_v, std_v, a_v, b_v = draw_values(
-            [self.mu, self.sigma, self.lower, self.upper], point=point, size=size)
+        mu_v, std_v, a_v, b_v = broadcast_distribution_samples(
+            draw_values(
+                [self.mu, self.sigma, self.lower, self.upper],
+                point=point,
+                size=size),
+            size=size,
+        )
         return generate_samples(stats.truncnorm.rvs,
                                 a=(a_v - mu_v)/std_v,
                                 b=(b_v - mu_v) / std_v,