diff --git a/src/lerobot/envs/configs.py b/src/lerobot/envs/configs.py index c9c43757d..af5bda33f 100644 --- a/src/lerobot/envs/configs.py +++ b/src/lerobot/envs/configs.py @@ -82,7 +82,7 @@ class EnvConfig(draccus.ChoiceRegistry, abc.ABC): def create_envs( self, n_envs: int, - use_async_envs: bool = True, + use_async_envs: bool = False, ) -> dict[str, dict[int, gym.vector.VectorEnv]]: """Create {suite: {task_id: VectorEnv}}. @@ -417,7 +417,7 @@ class LiberoEnv(EnvConfig): kwargs["task_ids"] = self.task_ids 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 if self.task is None: @@ -486,7 +486,7 @@ class MetaworldEnv(EnvConfig): "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 if self.task is None: diff --git a/src/lerobot/envs/factory.py b/src/lerobot/envs/factory.py index d349fef49..40d5425cc 100644 --- a/src/lerobot/envs/factory.py +++ b/src/lerobot/envs/factory.py @@ -58,7 +58,7 @@ def make_env_pre_post_processors( def make_env( cfg: EnvConfig | str, n_envs: int = 1, - use_async_envs: bool = True, + use_async_envs: bool = False, hub_cache_dir: str | None = None, trust_remote_code: bool = False, ) -> dict[str, dict[int, gym.vector.VectorEnv]]: diff --git a/tests/envs/test_dispatch.py b/tests/envs/test_dispatch.py index 8e2a93601..5bd2827f3 100644 --- a/tests/envs/test_dispatch.py +++ b/tests/envs/test_dispatch.py @@ -87,7 +87,7 @@ def test_base_create_envs(): return {} 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 env = envs["_dispatch_base_test"][0] assert isinstance(env, gym.vector.VectorEnv)