diff --git a/dipy/viz/horizon/util.py b/dipy/viz/horizon/util.py index 21304b92e7..69baba9dfe 100644 --- a/dipy/viz/horizon/util.py +++ b/dipy/viz/horizon/util.py @@ -86,4 +86,4 @@ def is_binary_image(data, unique_points=100): for dim in data.shape: indices.append(rng.integers(0, dim - 1, size=unique_points)) - return np.unique(data[*indices]).shape[0] <= 2 + return np.unique(np.take(data, indices)).shape[0] <= 2