wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Comment aloha, add test to CI

This commit is contained in:
Simon Alibert 2024-03-07 15:57:38 +01:00
parent 524d29aa80
commit 54b05bfb77
2 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
.github/workflows/test.yml vendored
View File

@ -87,6 +87,10 @@ jobs:
#---------------------------------------------- #----------------------------------------------
# run tests # run tests
#---------------------------------------------- #----------------------------------------------
- name: Run tests
run: |
source .venv/bin/activate
pytest tests
- name: Test train pusht end-to-end - name: Test train pusht end-to-end
run: | run: |
source .venv/bin/activate source .venv/bin/activate

9
tests/test_datasets.py
View File

@ -11,10 +11,11 @@ from .utils import init_config
# TODO(rcadene): simxarm is depreciated for now # TODO(rcadene): simxarm is depreciated for now
# ("simxarm", "lift"), # ("simxarm", "lift"),
("pusht", "pusht"), ("pusht", "pusht"),
("aloha", "sim_insertion_human"), # TODO(aliberts): add aloha when dataset is available on hub
("aloha", "sim_insertion_scripted"), # ("aloha", "sim_insertion_human"),
("aloha", "sim_transfer_cube_human"), # ("aloha", "sim_insertion_scripted"),
("aloha", "sim_transfer_cube_scripted"), # ("aloha", "sim_transfer_cube_human"),
# ("aloha", "sim_transfer_cube_scripted"),
], ],
) )
def test_factory(env_name, dataset_id): def test_factory(env_name, dataset_id):