diff --git a/lerobot/common/envs/factory.py b/lerobot/common/envs/factory.py index 83f94cfe..7932501d 100644 --- a/lerobot/common/envs/factory.py +++ b/lerobot/common/envs/factory.py @@ -28,11 +28,11 @@ def make_env(cfg: DictConfig, n_envs: int | None = None) -> gym.vector.VectorEnv raise ValueError("`n_envs must be at least 1") kwargs = { - "obs_type": "pixels_agent_pos", - "render_mode": "rgb_array", + # "obs_type": "pixels_agent_pos", + # "render_mode": "rgb_array", "max_episode_steps": cfg.env.episode_length, - "visualization_width": 384, - "visualization_height": 384, + # "visualization_width": 384, + # "visualization_height": 384, } package_name = f"gym_{cfg.env.name}" diff --git a/lerobot/configs/env/dora.yaml b/lerobot/configs/env/dora.yaml new file mode 100644 index 00000000..33b49973 --- /dev/null +++ b/lerobot/configs/env/dora.yaml @@ -0,0 +1,14 @@ +# @package _global_ + +fps: 30 + +env: + name: dora + task: DoraAloha-v0 + # from_pixels: True + # pixels_only: False + # image_size: [3, 480, 640] + episode_length: 400 + # fps: ${fps} + # state_dim: 14 + # action_dim: 14