hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Integrate pusht env from diffusion

Browse Source
This commit is contained in:
Simon Alibert
2024-03-10 16:33:03 +01:00
parent 302b78962c
commit 6c867d78ef
10 changed files with 801 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_envs.py
View File

@@ -3,7 +3,7 @@ from tensordict import TensorDict
from torchrl.envs.utils import check_env_specs, step_mdp
from lerobot.common.envs.factory import make_env
from lerobot.common.envs.pusht import PushtEnv
from lerobot.common.envs.pusht.pusht import PushtEnv
from lerobot.common.envs.simxarm import SimxarmEnv
from .utils import init_config