Merge branch 'vectorized_env' of github.com:nicklashansen/tdmpc2 into vectorized_env

2025-05-20 13:44:41 -07:00
parent 10f368f20d 10a0be2724
commit 491d367fc6
1 changed files with 3 additions and 0 deletions
--- 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):
 	"""