Update conversion script to shuffle deterministically

apockill · Jan 7, 2025 · c810242 · c810242
1 parent cb25789
commit c810242
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkgs/myarm_ai/myarm_ai/scripts/convert_isaac_to_lerobot.py b/pkgs/myarm_ai/myarm_ai/scripts/convert_isaac_to_lerobot.py
@@ -21,7 +21,7 @@ def convert(input_dataset: Path, repo_id: str, fps: int, task: str) -> LeRobotDa
     shuffled_renders = list(traj_dataset.renders)
 
     # Deterministic shuffle
-    shuffled_renders.sort()
+    shuffled_renders.sort(key=lambda x: x.episode_json_path)
     random.seed(1337)
     random.shuffle(shuffled_renders)