diff --git a/fancy_gym/envs/mujoco/air_hockey/air_hockey_env_wrapper.py b/fancy_gym/envs/mujoco/air_hockey/air_hockey_env_wrapper.py index b5ef8c3..31a1a8e 100644 --- a/fancy_gym/envs/mujoco/air_hockey/air_hockey_env_wrapper.py +++ b/fancy_gym/envs/mujoco/air_hockey/air_hockey_env_wrapper.py @@ -130,6 +130,7 @@ class AirHockeyEnv(Environment): return self.base_env.render(record = True) elif self.render_mode == 'human': self.render_human_active = True + self.base_env.render() else: raise ValueError(f"Unsupported render mode: '{self.render_mode}'") @@ -190,4 +191,4 @@ if __name__ == "__main__": J = 0. gamma = 1. steps = 0 - env.reset() \ No newline at end of file + env.reset()