Fix tests

2024-11-20 00:26:31 +01:00
parent 9ee8711504
commit f43e5d07f5
2 changed files with 12 additions and 5 deletions
--- a/lerobot/common/datasets/utils.py
+++ b/lerobot/common/datasets/utils.py
@@ -136,7 +136,10 @@ def append_jsonlines(data: dict, fpath: Path) -> None:


 def load_info(local_dir: Path) -> dict:
-    return load_json(local_dir / INFO_PATH)
+    info = load_json(local_dir / INFO_PATH)
+    for ft in info["features"].values():
+        ft["shape"] = tuple(ft["shape"])
+    return info


 def load_stats(local_dir: Path) -> dict: