Merge pull request #270 from eteq/fix-np15-error

Fix for numpy 1.15
dfm · Aug 14, 2018 · 8a473d5 · 8a473d5
2 parents 2b171ae + cbe3c25
commit 8a473d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/emcee/ensemble.py b/emcee/ensemble.py
@@ -406,7 +406,7 @@ def compute_log_prob(self, coords):
             if len(shape):
                 axes = np.arange(len(shape))[np.array(shape) == 1] + 1
                 if len(axes):
-                    blob = np.squeeze(blob, axes)
+                    blob = np.squeeze(blob, tuple(axes))
 
         # Check for log_prob returning NaN.
         if np.any(np.isnan(log_prob)):