revision 1

2024-04-15 10:56:43 +01:00
parent 40d417ef60
commit 30023535f9
6 changed files with 24 additions and 20 deletions
--- a/lerobot/common/datasets/factory.py
+++ b/lerobot/common/datasets/factory.py
@@ -86,10 +86,11 @@ def make_dataset(
            ]
        )

-    delta_timestamps = cfg.policy.delta_timestamps
-    for key in delta_timestamps:
-        if isinstance(delta_timestamps[key], str):
-            delta_timestamps[key] = eval(delta_timestamps[key])
+    delta_timestamps = cfg.policy.get("delta_timestamps")
+    if delta_timestamps is not None:
+        for key in delta_timestamps:
+            if isinstance(delta_timestamps[key], str):
+                delta_timestamps[key] = eval(delta_timestamps[key])

    dataset = clsfunc(
        dataset_id=cfg.dataset_id,