fixes shape tests with new scalar behavior

michaelosthege · michaelosthege · commit 058920345365 · 2020-11-09T17:09:50.000+01:00
diff --git a/pymc3/distributions/discrete.py b/pymc3/distributions/discrete.py
@@ -972,7 +972,7 @@ def random(self, point=None, size=None):
         return generate_samples(
             random_choice,
             p=p,
-            broadcast_shape=p.shape[:-1] or (1,),
+            broadcast_shape=p.shape[:-1],
             dist_shape=self.shape,
             size=size,
         )

Original file line number	Diff line number	Diff line change
`@@ -972,7 +972,7 @@ def random(self, point=None, size=None):`
`972`	`972`	`return generate_samples(`
`973`	`973`	`random_choice,`
`974`	`974`	`p=p,`
`975`		`- broadcast_shape=p.shape[:-1] or (1,),`
	`975`	`+ broadcast_shape=p.shape[:-1],`
`976`	`976`	`dist_shape=self.shape,`
`977`	`977`	`size=size,`
`978`	`978`	`)`