tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Add simxarm back into tests

Browse Source
This commit is contained in:
Simon Alibert
2024-03-25 16:35:46 +01:00
parent d3adaf1379
commit 058ac991eb
18 changed files with 44 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tests/test_envs.py
View File

@@ -39,19 +39,20 @@ def print_spec_rollout(env):
print("data from rollout:", simple_rollout(100))
@pytest.mark.skip(reason="Simxarm is deprecated")
# @pytest.mark.skip(reason="Simxarm is deprecated")
@pytest.mark.parametrize(
"task,from_pixels,pixels_only",
[
("lift", False, False),
("lift", True, False),
("lift", True, True),
("reach", False, False),
("reach", True, False),
("push", False, False),
("push", True, False),
("peg_in_box", False, False),
("peg_in_box", True, False),
# TODO(aliberts): Add simxarm other task or remove them completely from repo
# ("reach", False, False),
# ("reach", True, False),
# ("push", False, False),
# ("push", True, False),
# ("peg_in_box", False, False),
# ("peg_in_box", True, False),
],
)
def test_simxarm(task, from_pixels, pixels_only):
@@ -84,7 +85,7 @@ def test_pusht(from_pixels, pixels_only):
@pytest.mark.parametrize(
"env_name",
[
# "simxarm",
"simxarm",
"pusht",
"aloha",
],