fix destructuring

AI-Hypercomputer · Nov 9, 2024 · 2d2ddea · 2d2ddea
1 parent 3d3eaed
commit 2d2ddea
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
@@ -144,7 +144,7 @@ async def deserialize(
     results = [None] * len(infos)
 
     logging.warning(f"[ksadi] restore loop will take {len(inputs_by_global_mesh.items())} iterations")
-    for i, global_mesh, idxs in enumerate(inputs_by_global_mesh.items()):
+    for i, (global_mesh, idxs) in enumerate(inputs_by_global_mesh.items()):
       grouped_infos = [infos[idx] for idx in idxs]
       grouped_global_shapes = [global_shapes[idx] for idx in idxs]
       grouped_dtypes = [dtypes[idx] for idx in idxs]