Add regression tests (#119)

- Add `tests/scripts/save_policy_to_safetensor.py` to generate test artifacts - Add `test_backward_compatibility to test generated outputs from the policies against artifacts
2024-05-04 16:20:30 +02:00
parent 19812ca470
commit c77633c38c
15 changed files with 236 additions and 43 deletions
--- a/lerobot/configs/policy/tdmpc.yaml
+++ b/lerobot/configs/policy/tdmpc.yaml
@@ -1,6 +1,7 @@
 # @package _global_

 seed: 1
+dataset_repo_id: lerobot/xarm_lift_medium_replay

 training:
  offline_steps: 25000