rm unused

AI-Hypercomputer · Nov 27, 2024 · 1173f9b · 1173f9b
1 parent 4795c97
commit 1173f9b
Showing 1 changed file with 10 additions and 8 deletions.
diff --git 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 = [