From c91ececc7546c0c9cf8051cff26b2d6eb02552ea Mon Sep 17 00:00:00 2001
From: Remi Cadene <re.cadene@gmail.com>
Date: Fri, 24 May 2024 09:06:29 +0000
Subject: [PATCH] WIP

---
 lerobot/common/envs/factory.py |  8 ++++----
 lerobot/configs/env/dora.yaml  | 14 ++++++++++++++
 2 files changed, 18 insertions(+), 4 deletions(-)
 create mode 100644 lerobot/configs/env/dora.yaml

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