make skip public

Kautenja · Jan 6, 2019 · c198ee9 · c198ee9
1 parent 73daa27
commit c198ee9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nes_py/wrappers/frame_skip_env.py b/nes_py/wrappers/frame_skip_env.py
@@ -18,7 +18,7 @@ def __init__(self, env, skip=4):
 
         """
         super(FrameSkipEnv, self).__init__(env)
-        self._skip = skip
+        self.skip = skip
 
     def step(self, action):
         """
@@ -40,7 +40,7 @@ def step(self, action):
         # set the done flag to false
         done = False
         # iterate over the number of frames to skip
-        for _ in range(self._skip):
+        for _ in range(self.skip):
             # get the next state using the given action
             state, reward, done, info = self.env.step(action)
             # accumulate the total reward