use starmap

AI-Hypercomputer · Nov 27, 2024 · f4dd5db · f4dd5db
1 parent 4da7b43
commit f4dd5db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pathwaysutils/persistence/pathways_orbax_handler.py b/pathwaysutils/persistence/pathways_orbax_handler.py
@@ -176,7 +176,7 @@ async def deserialize(
                 )
             ]
             with Pool() as p:
-                grouped_arrays_and_futures = p.map(f, *args_list)
+                grouped_arrays_and_futures = p.starmap(f, args_list)
             # grouped_arrays_and_futures = [
             #     f(
             #         location=location,