admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-20 19:19:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

add task ids to libero env cfg (#2842)

Browse Source
This commit is contained in:
Reece O'Mahoney
2026-01-26 16:26:49 +00:00
committed by GitHub
parent 9e10eb4a77
commit 366bef915c
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/envs/configs.py
+5 -4
View File
@@ -260,6 +260,7 @@ class HILSerlRobotEnvConfig(EnvConfig):
@dataclass @dataclass
class LiberoEnv(EnvConfig): class LiberoEnv(EnvConfig):
task: str = "libero_10" # can also choose libero_spatial, libero_object, etc. task: str = "libero_10" # can also choose libero_spatial, libero_object, etc.
task_ids: list[int] | None = None
fps: int = 30 fps: int = 30
episode_length: int | None = None episode_length: int | None = None
obs_type: str = "pixels_agent_pos" obs_type: str = "pixels_agent_pos"
@@ -338,10 +339,10 @@ class LiberoEnv(EnvConfig):
@property @property
def gym_kwargs(self) -> dict: def gym_kwargs(self) -> dict:
return { kwargs: dict[str, Any] = {"obs_type": self.obs_type, "render_mode": self.render_mode}
"obs_type": self.obs_type, if self.task_ids is not None:
"render_mode": self.render_mode, kwargs["task_ids"] = self.task_ids
} return kwargs
@EnvConfig.register_subclass("metaworld") @EnvConfig.register_subclass("metaworld")