From 7432fc2d8c24f93edf0525c764a03a2e0a20be90 Mon Sep 17 00:00:00 2001 From: Daniel Tan Date: Wed, 23 Aug 2023 21:33:57 +0100 Subject: [PATCH] Enable AutoReset for Mujoco envs --- omnisafe/envs/mujoco_env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/omnisafe/envs/mujoco_env.py b/omnisafe/envs/mujoco_env.py index cdedfb29a..f5ba67f53 100644 --- a/omnisafe/envs/mujoco_env.py +++ b/omnisafe/envs/mujoco_env.py @@ -35,7 +35,7 @@ class MujocoEnv(CMDP): need_time_limit_wrapper (bool): Whether to use time limit wrapper. """ - need_auto_reset_wrapper = False + need_auto_reset_wrapper = True need_time_limit_wrapper = False need_action_repeat_wrapper = True