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

rename handle -> task

Browse Source
This commit is contained in:
Cadene
2024-04-08 14:54:52 +00:00
parent 6c792f0d3d
commit 1149894e1d
6 changed files with 12 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_envs.py
View File

@@ -14,7 +14,7 @@ from .utils import DEVICE, DEFAULT_CONFIG_PATH
@pytest.mark.parametrize(
"env_name, handle, obs_type",
"env_name, task, obs_type",
[
# ("AlohaInsertion-v0", "state"),
("aloha", "AlohaInsertion-v0", "pixels"),
@@ -29,10 +29,10 @@ from .utils import DEVICE, DEFAULT_CONFIG_PATH
("pusht", "PushT-v0", "pixels_agent_pos"),
],
)
def test_env(env_name, handle, obs_type):
def test_env(env_name, task, obs_type):
package_name = f"gym_{env_name}"
importlib.import_module(package_name)
env = gym.make(f"{package_name}/{handle}", obs_type=obs_type)
env = gym.make(f"{package_name}/{task}", obs_type=obs_type)
check_env(env.unwrapped)