mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-21 03:30:10 +00:00
chore: revert configs.py, factory.py, test_dispatch.py to main
These use_async_envs default changes belong to the async-vector-env PR (#3274), not this CI PR. Restore to match origin/main. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -82,7 +82,7 @@ class EnvConfig(draccus.ChoiceRegistry, abc.ABC):
|
|||||||
def create_envs(
|
def create_envs(
|
||||||
self,
|
self,
|
||||||
n_envs: int,
|
n_envs: int,
|
||||||
use_async_envs: bool = True,
|
use_async_envs: bool = False,
|
||||||
) -> dict[str, dict[int, gym.vector.VectorEnv]]:
|
) -> dict[str, dict[int, gym.vector.VectorEnv]]:
|
||||||
"""Create {suite: {task_id: VectorEnv}}.
|
"""Create {suite: {task_id: VectorEnv}}.
|
||||||
|
|
||||||
@@ -417,7 +417,7 @@ class LiberoEnv(EnvConfig):
|
|||||||
kwargs["task_ids"] = self.task_ids
|
kwargs["task_ids"] = self.task_ids
|
||||||
return kwargs
|
return kwargs
|
||||||
|
|
||||||
def create_envs(self, n_envs: int, use_async_envs: bool = True):
|
def create_envs(self, n_envs: int, use_async_envs: bool = False):
|
||||||
from lerobot.envs.libero import create_libero_envs
|
from lerobot.envs.libero import create_libero_envs
|
||||||
|
|
||||||
if self.task is None:
|
if self.task is None:
|
||||||
@@ -486,7 +486,7 @@ class MetaworldEnv(EnvConfig):
|
|||||||
"render_mode": self.render_mode,
|
"render_mode": self.render_mode,
|
||||||
}
|
}
|
||||||
|
|
||||||
def create_envs(self, n_envs: int, use_async_envs: bool = True):
|
def create_envs(self, n_envs: int, use_async_envs: bool = False):
|
||||||
from lerobot.envs.metaworld import create_metaworld_envs
|
from lerobot.envs.metaworld import create_metaworld_envs
|
||||||
|
|
||||||
if self.task is None:
|
if self.task is None:
|
||||||
|
|||||||
@@ -58,7 +58,7 @@ def make_env_pre_post_processors(
|
|||||||
def make_env(
|
def make_env(
|
||||||
cfg: EnvConfig | str,
|
cfg: EnvConfig | str,
|
||||||
n_envs: int = 1,
|
n_envs: int = 1,
|
||||||
use_async_envs: bool = True,
|
use_async_envs: bool = False,
|
||||||
hub_cache_dir: str | None = None,
|
hub_cache_dir: str | None = None,
|
||||||
trust_remote_code: bool = False,
|
trust_remote_code: bool = False,
|
||||||
) -> dict[str, dict[int, gym.vector.VectorEnv]]:
|
) -> dict[str, dict[int, gym.vector.VectorEnv]]:
|
||||||
|
|||||||
@@ -87,7 +87,7 @@ def test_base_create_envs():
|
|||||||
return {}
|
return {}
|
||||||
|
|
||||||
try:
|
try:
|
||||||
envs = _Env().create_envs(n_envs=2, use_async_envs=False)
|
envs = _Env().create_envs(n_envs=2)
|
||||||
assert "_dispatch_base_test" in envs
|
assert "_dispatch_base_test" in envs
|
||||||
env = envs["_dispatch_base_test"][0]
|
env = envs["_dispatch_base_test"][0]
|
||||||
assert isinstance(env, gym.vector.VectorEnv)
|
assert isinstance(env, gym.vector.VectorEnv)
|
||||||
|
|||||||
Reference in New Issue
Block a user