Skip to content

Commit 0150b75

Browse files
author
colin
committed
Fix python2 build errors
1 parent e7dc271 commit 0150b75

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc3/tests/test_distributions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def product(domains, n_samples=-1):
8080
return []
8181
all_vals = [zip(names, val) for val in itertools.product(*[d.vals for d in domains])]
8282
if n_samples > 0 and len(all_vals) > n_samples:
83-
return nr.choice(all_vals, n_samples, replace=False)
83+
return nr.choice(np.atleast_1d(all_vals), n_samples, replace=False)
8484
return all_vals
8585

8686

0 commit comments

Comments
 (0)