Set minimum version for NumPy requirement (#532)

* Add minimum NumPy version * Revert "Convert array of axes to list of Python scalars" This reverts commit e6f8e9c.
fepegar · May 12, 2021 · 4b766ea · 4b766ea
1 parent c7af603
commit 4b766ea
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -5,7 +5,7 @@ furo
 humanize
 matplotlib
 nibabel
-numpy
+numpy>=1.15
 scipy
 sphinx-copybutton
 sphinx-gallery

diff --git a/setup.py b/setup.py
@@ -15,7 +15,7 @@
     'Deprecated',
     'humanize',
     'nibabel',
-    'numpy',
+    'numpy>=1.15',
     'scipy',
     'torch>=1.1',
     'tqdm',

diff --git a/torchio/transforms/augmentation/spatial/random_flip.py b/torchio/transforms/augmentation/spatial/random_flip.py
@@ -125,7 +125,7 @@ def _ensure_axes_indices(subject, axes):
 def _flip_image(image, axes):
     spatial_axes = np.array(axes, int) + 1
     data = image.numpy()
-    data = np.flip(data, axis=spatial_axes.tolist())
+    data = np.flip(data, axis=spatial_axes)
     data = data.copy()  # remove negative strides
     data = torch.as_tensor(data)
     image.set_data(data)