Use correct index in masking (mainly for soap).

SWIFTSIM · Oct 10, 2024 · 973ce95 · 973ce95
1 parent c3087ee
commit 973ce95
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/swiftgalaxy/halo_catalogues.py b/swiftgalaxy/halo_catalogues.py
@@ -350,8 +350,8 @@ def __getattr__(self, attr: str) -> Any:
             except AttributeError:
                 return None
         obj = getattr(self._catalogue, attr)
-        if self._multi_galaxy_catalogue_mask is not None:
-            return _MaskHelper(obj, self._multi_galaxy_catalogue_mask)
+        if self._multi_galaxy_index_mask is not None:
+            return _MaskHelper(obj, self._multi_galaxy_index_mask)
         else:
             return obj
 
@@ -834,8 +834,8 @@ def centre(self) -> cosmo_array:
         obj = self._catalogue
         for attr in self.centre_type.split("."):
             obj = getattr(obj, attr)
-        if self._multi_galaxy_catalogue_mask is not None:
-            return obj[self._multi_galaxy_catalogue_mask]
+        if self._multi_galaxy_index_mask is not None:
+            return obj[self._multi_galaxy_index_mask]
         return obj.squeeze()
 
     @property
@@ -855,8 +855,8 @@ def velocity_centre(self) -> cosmo_array:
         obj = self._catalogue
         for attr in self.velocity_centre_type.split("."):
             obj = getattr(obj, attr)
-        if self._multi_galaxy_catalogue_mask is not None:
-            return obj[self._multi_galaxy_catalogue_mask]
+        if self._multi_galaxy_index_mask is not None:
+            return obj[self._multi_galaxy_index_mask]
         return obj.squeeze()
 
     def __repr__(self) -> str: