Use a more robust approach to static shape inference in Dirichlet

brandonwillard · brandonwillard · commit 7e57248e10f2 · 2020-09-01T15:27:51.000-05:00
diff --git a/pymc3/distributions/multivariate.py b/pymc3/distributions/multivariate.py
@@ -498,7 +498,7 @@ def __init__(self, a, transform=transforms.stick_breaking,
                 DeprecationWarning
             )
             try:
-                kwargs['shape'] = get_test_value(tt.shape(a))
+                kwargs['shape'] = np.shape(get_test_value(a))
             except AttributeError:
                 pass
 

Original file line number	Diff line number	Diff line change
`@@ -498,7 +498,7 @@ def __init__(self, a, transform=transforms.stick_breaking,`
`498`	`498`	`DeprecationWarning`
`499`	`499`	`)`
`500`	`500`	`try:`
`501`		`- kwargs['shape'] = get_test_value(tt.shape(a))`
	`501`	`+ kwargs['shape'] = np.shape(get_test_value(a))`
`502`	`502`	`except AttributeError:`
`503`	`503`	`pass`
`504`	`504`