diff --git a/pathwaysutils/persistence/pathways_orbax_handler.py b/pathwaysutils/persistence/pathways_orbax_handler.py index 82b2d6b..80bfcdb 100644 --- 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,