From 1173f9b12e3016bd4b9fb0c84d9cc4e47a0b7c4d Mon Sep 17 00:00:00 2001 From: Sadi Kneipp Date: Wed, 27 Nov 2024 22:11:22 +0000 Subject: [PATCH] rm unused --- .../persistence/pathways_orbax_handler.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/pathwaysutils/persistence/pathways_orbax_handler.py b/pathwaysutils/persistence/pathways_orbax_handler.py index fe32247..670d915 100644 --- a/pathwaysutils/persistence/pathways_orbax_handler.py +++ b/pathwaysutils/persistence/pathways_orbax_handler.py @@ -165,14 +165,16 @@ async def deserialize( ) grouped_arrays_and_futures = None - from functools import partial - args_list = [[location, name, dtype, shape, sharding] for location, name, dtype, shape, sharding in zip( - locations, - names, - grouped_dtypes, - grouped_global_shapes, - grouped_shardings, - )] + args_list = [ + [location, name, dtype, shape, sharding] + for location, name, dtype, shape, sharding in zip( + locations, + names, + grouped_dtypes, + grouped_global_shapes, + grouped_shardings, + ) + ] with Pool() as p: grouped_arrays_and_futures = p.apply(f, args_list) # grouped_arrays_and_futures = [