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