From d44e10fca3f89c9e0976b8c7bcbd52a3c8c00e16 Mon Sep 17 00:00:00 2001
From: = <=>
Date: Fri, 22 Nov 2024 21:05:57 +0100
Subject: [PATCH] only load dict
---
src/cyxtrax/io/load_maps.py | 11 ++++++++---
1 file changed, 8 insertions(+), 3 deletions(-)
diff --git a/src/cyxtrax/io/load_maps.py b/src/cyxtrax/io/load_maps.py
index 3364018..054578a 100644
--- a/src/cyxtrax/io/load_maps.py
+++ b/src/cyxtrax/io/load_maps.py
@@ -13,9 +13,14 @@ def load_atlas(load_path: Path) -> tuple[jnp.ndarray, jnp.ndarray, list[MeshObje
with h5py.File(load_path, "r") as f:
maps = jnp.array(f["/maps"][:])
points = jnp.array(f["/positions"][:])
+
+ mesh_object_dict = load_atlas_dict(load_path)
+ mesh_object_list = list(map(from_dict, mesh_object_dict))
+ return maps, points, mesh_object_list
+
+def load_atlas_dict(load_path: Path):
+ with h5py.File(load_path, "r") as f:
mesh_object_str = f.attrs["mesh_list"]
mesh_object_dict = json.loads(mesh_object_str)
- mesh_object_list = list(map(from_dict, mesh_object_dict))
-
- return maps, points, mesh_object_list
+ return mesh_object_dict
--
GitLab