Add gym-aloha, rename simxarm -> xarm, refactor

2024-04-08 16:18:53 +02:00
parent 5dff6d8339
commit 3f6dfa4916
15 changed files with 91 additions and 97 deletions
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -204,7 +204,7 @@ jobs:
          source .venv/bin/activate
          python lerobot/scripts/train.py \
            policy=tdmpc \
-            env=simxarm \
+            env=xarm \
            wandb.enable=False \
            offline_steps=1 \
            online_steps=1 \
@@ -229,6 +229,6 @@ jobs:
          python lerobot/scripts/eval.py \
            --config lerobot/configs/default.yaml \
            policy=tdmpc  \
-            env=simxarm \
+            env=xarm \
            eval_episodes=1 \
            device=cpu