RF: No longer need to support numpy 1.6

dipy · May 8, 2017 · 9fa1f4b · 9fa1f4b
1 parent df9c160
commit 9fa1f4b
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/dipy/tracking/utils.py b/dipy/tracking/utils.py
@@ -240,11 +240,8 @@ def ndbincount(x, weights=None, shape=None):
         shape = x.max(1) + 1
 
     x = ravel_multi_index(x, shape)
-    # out = np.bincount(x, weights, minlength=np.prod(shape))
-    # out.shape = shape
-    # Use resize to be compatible with numpy < 1.6, minlength new in 1.6
-    out = np.bincount(x, weights)
-    out.resize(shape)
+    out = np.bincount(x, weights, minlength=np.prod(shape))
+    out.shape = shape
 
     return out