From 10f368f20de1fcd0030f6c86b03ed779ff9022b8 Mon Sep 17 00:00:00 2001 From: Nicklas Hansen Date: Sun, 11 Feb 2024 14:41:20 -0800 Subject: [PATCH] init --- tdmpc2/envs/dmcontrol.py | 3 +++ 1 file changed, 3 insertions(+) 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): """