Revert "Convert array of axes to list of Python scalars"

This reverts commit e6f8e9c.
fepegar · May 12, 2021 · 5c53b2c · 5c53b2c
1 parent c0fd895
commit 5c53b2c
Showing 1 changed file with 1 addition and 1 deletion.
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)