diff --git a/tdmpc2/envs/dmcontrol.py b/tdmpc2/envs/dmcontrol.py index b8c9d2a..695c3ae 100644 --- a/tdmpc2/envs/dmcontrol.py +++ b/tdmpc2/envs/dmcontrol.py @@ -101,6 +101,9 @@ class Pixels(gym.Wrapper): def close(self): self.env.close() + def close(self): + self.env.close() + def make_env(cfg): """