Explicity cast start, step to int in regular_grid

This commit is contained in:
Juan Nunez-Iglesias
2013-12-13 00:54:57 +11:00
parent 0d5b822110
commit a67c5e1e2b
+2 -2
View File
@@ -64,8 +64,8 @@ def regular_grid(ar_shape, n_points):
(1.0 / (ndim - dim - 1))) (1.0 / (ndim - dim - 1)))
if (sorted_dims >= stepsizes).all(): if (sorted_dims >= stepsizes).all():
break break
starts = stepsizes // 2 starts = (stepsizes // 2).astype(int)
stepsizes = np.round(stepsizes) stepsizes = np.round(stepsizes).astype(int)
slices = [slice(start, None, step) for slices = [slice(start, None, step) for
start, step in zip(starts, stepsizes)] start, step in zip(starts, stepsizes)]
slices = [slices[i] for i in unsort_dim_idxs] slices = [slices[i] for i in unsort_dim_idxs]