fix action repeat wrapper (#67)

danijar · Jun 14, 2023 · 8fa35f8 · 8fa35f8
1 parent 423291a
commit 8fa35f8
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/dreamerv3/embodied/core/wrappers.py b/dreamerv3/embodied/core/wrappers.py
@@ -41,10 +41,9 @@ class ActionRepeat(base.Wrapper):
   def __init__(self, env, repeat):
     super().__init__(env)
     self._repeat = repeat
-    self._done = False
 
   def step(self, action):
-    if action['reset'] or self._done:
+    if action['reset']:
       return self.env.step(action)
     reward = 0.0
     for _ in range(self._repeat):
@@ -53,7 +52,6 @@ def step(self, action):
       if obs['is_last'] or obs['is_terminal']:
         break
     obs['reward'] = np.float32(reward)
-    self._done = obs['is_last']
     return obs