fix concat states

2026-06-27 16:46:41 +08:00 · 2017-11-12 14:16:49 +08:00
parent 2e5a154030
commit 4b3701ceb6
1 changed files with 4 additions and 0 deletions
@@ -38,3 +38,7 @@ class ConcatStates(gym.Wrapper):
        state = concat_states(state)

        return state, reward, done, info
+
+    def reset(self):
+        state = self.env.reset()
+        return concat_states(state)