implement reset

Kautenja · Jul 22, 2018 · 8efb860 · 8efb860
1 parent 1904da1
commit 8efb860
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/nes_py/wrappers/binary_to_discrete_space_env.py b/nes_py/wrappers/binary_to_discrete_space_env.py
@@ -65,6 +65,9 @@ def step(self, action):
         # take the step and record the output
         return self.env.step(self._action_map[action])
 
+    def reset(self):
+        """Reset the environment and return the initial observation."""
+        return self.env.reset()
 
 # explicitly define the outward facing API of this module
 __all__ = [BinarySpaceToDiscreteSpaceEnv.__name__]