Add close method to AtariPreprocessing.

PiperOrigin-RevId: 235068759
ribeirojose · Apr 11, 2019 · 00a5b32 · 00a5b32
1 parent 711e188
commit 00a5b32
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/dopamine/discrete_domains/atari_lib.py b/dopamine/discrete_domains/atari_lib.py
@@ -276,6 +276,9 @@ def reward_range(self):
   def metadata(self):
     return self.environment.metadata
 
+  def close(self):
+    return self.environment.close()
+
   def reset(self):
     """Resets the environment.