Skip to content

Commit e94e4f1

Browse files
committed
Add check if zerosum_axes is iterable in dist method
1 parent fd3aefa commit e94e4f1

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pymc/distributions/multivariate.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2463,6 +2463,8 @@ def __new__(cls, *args, zerosum_axes=None, dims=None, **kwargs):
24632463
def dist(cls, sigma=1, zerosum_axes=None, **kwargs):
24642464
if zerosum_axes is None:
24652465
zerosum_axes = [-1]
2466+
if not isinstance(zerosum_axes, (list, tuple)):
2467+
zerosum_axes = [zerosum_axes]
24662468

24672469
sigma = at.as_tensor_variable(floatX(sigma))
24682470
if sigma.ndim > 0:

0 commit comments

Comments
 (0)