Update constrain_index for _shared mask.

SWIFTSIM · Sep 19, 2024 · 702e0ba · 702e0ba
1 parent 700ab97
commit 702e0ba
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/swiftsimio/masks.py b/swiftsimio/masks.py
@@ -521,9 +521,9 @@ def constrain_index(self, index: int):
         if not self.metadata.filetype == "SOAP":
             warnings.warn("Not masking a SOAP catalogue, nothing constrained.")
             return
-        for group_name in self.metadata.present_group_names:
-            setattr(self, group_name, np.array([[index, index + 1]]))
-            setattr(self, f"{group_name}_size", 1)
+        self._shared = np.array([[index, index + 1]])
+        self._shared_size = 1
+
         return
 
     def get_masked_counts_offsets(