init

2024-02-11 14:41:20 -08:00
parent 829e329b3b
commit 10f368f20d
1 changed files with 3 additions and 0 deletions
--- 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):
 	"""