update

dgriff777 · Nov 1, 2018 · eb5c9b9 · eb5c9b9
1 parent 474f480
commit eb5c9b9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/environment.py b/environment.py
@@ -180,10 +180,10 @@ def step(self, action):
 
         return max_frame, total_reward, done, info
 
-    def reset(self):
+    def reset(self, **kwargs):
         """Clear past frame buffer and init. to first obs. from inner env."""
         self._obs_buffer.clear()
-        obs = self.env.reset()
+        obs = self.env.reset(**kwargs)
         self._obs_buffer.append(obs)
         return obs