From d9019d9e7eae22b3b250ae445f35cae458c82464 Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Tue, 9 Apr 2024 10:24:28 +0200 Subject: [PATCH] disable env_checker in factory --- lerobot/common/envs/factory.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/lerobot/common/envs/factory.py b/lerobot/common/envs/factory.py index c8d10851..d5571935 100644 --- a/lerobot/common/envs/factory.py +++ b/lerobot/common/envs/factory.py @@ -30,10 +30,13 @@ def make_env(cfg, num_parallel_envs=0) -> gym.Env | gym.vector.SyncVectorEnv: if num_parallel_envs == 0: # non-batched version of the env that returns an observation of shape (c) - env = gym.make(gym_handle, **kwargs) + env = gym.make(gym_handle, disable_env_checker=True, **kwargs) else: # batched version of the env that returns an observation of shape (b, c) env = gym.vector.SyncVectorEnv( - [lambda: gym.make(gym_handle, **kwargs) for _ in range(num_parallel_envs)] + [ + lambda: gym.make(gym_handle, disable_env_checker=True, **kwargs) + for _ in range(num_parallel_envs) + ] ) return env