Get the args order right in map...

SWIFTSIM · Oct 14, 2024 · a91f50b · a91f50b
1 parent d9c2b46
commit a91f50b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/swiftgalaxy/iterator.py b/swiftgalaxy/iterator.py
@@ -755,10 +755,10 @@ def dm_median_position(
             args = [tuple()] * len(self.iteration_order)
         if kwargs is None:
             kwargs = [dict()] * len(self.iteration_order)
-        for sg in self:
+        for sg, iteration_location in zip(self, self.iteration_order):
             result[sg.halo_catalogue._multi_galaxy_catalogue_mask] = func(
                 sg,
-                args[sg.halo_catalogue._multi_galaxy_catalogue_mask],
-                kwargs[sg.halo_catalogue._multi_galaxy_catalogue_mask],
+                *args[iteration_location],
+                **kwargs[iteration_location],
             )
         return result