Fixed bug in logic for spatial_only in subcube_slices_from_mask

radio-astro-tools · Apr 28, 2020 · a03b5f5 · a03b5f5
1 parent cad11f4
commit a03b5f5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spectral_cube/dask_spectral_cube.py b/spectral_cube/dask_spectral_cube.py
@@ -860,7 +860,7 @@ def subcube_slices_from_mask(self, region_mask, spatial_only=False):
 
         slices = []
         for axis in range(3):
-            if spatial_only:
+            if axis == 0 and spatial_only:
                 slices.append(slice(None))
                 continue
             collapse_axes = tuple(index for index in range(3) if index != axis)