Skip to content

Commit e69ef7a

Browse files
author
Christopher Fonnesbeck
committed
Fix for IndexError in metropolis discrete indexing
1 parent bf8b8d8 commit e69ef7a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc3/step_methods/metropolis.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def __init__(self, vars=None, S=None, proposal_dist=NormalProposal, scaling=1.,
9191
self.accepted = 0
9292

9393
# Determine type of variables
94-
self.discrete = np.hstack(np.ravel([[v.dtype in discrete_types ] * v.dsize for v in vars]))
94+
self.discrete = np.ravel([[v.dtype in discrete_types ] * (v.dsize or 1) for v in vars])
9595
self.any_discrete = self.discrete.any()
9696
self.all_discrete = self.discrete.all()
9797

0 commit comments

Comments
 (0)