fix typo

AI-Hypercomputer · Dec 3, 2024 · d12a41e · d12a41e
1 parent ee48797
commit d12a41e
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
@@ -155,7 +155,7 @@ async def deserialize(
             grouped_shardings = [shardings[idx] for idx in idxs]
             locations, names = extract_parent_dir_and_name(grouped_infos)
             grouped_arrays = helper.read_arrays(
-                location[0],
+                locations[0],
                 names,
                 grouped_dtypes,
                 grouped_global_shapes,