Fix datasets missing versions (#318)

2024-07-16 23:02:31 +02:00
parent 5f5efe7cb9
commit 8865e19c12
12 changed files with 156 additions and 120 deletions
--- a/lerobot/scripts/control_robot.py
+++ b/lerobot/scripts/control_robot.py
@@ -475,6 +475,7 @@ def record_dataset(
    hf_dataset = to_hf_dataset(data_dict, video)
    episode_data_index = calculate_episode_data_index(hf_dataset)
    info = {
+        "codebase_version": CODEBASE_VERSION,
        "fps": fps,
        "video": video,
    }