fix pipelining

AI-Hypercomputer · Nov 27, 2024 · 4da7b43 · 4da7b43
1 parent d9ae4fd
commit 4da7b43
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.map(f, *args_list)
             # grouped_arrays_and_futures = [
             #     f(
             #         location=location,