(self)
| 89 | self.k = k |
| 90 | |
| 91 | def reset(self): |
| 92 | # This assumes that reset() will really reset the env. |
| 93 | # If the underlying env tries to be smart about reset |
| 94 | # (e.g. end-of-life), the assumption doesn't hold. |
| 95 | ob = self.env.reset() |
| 96 | self.cnt = 0 |
| 97 | return ob |
| 98 | |
| 99 | def step(self, action): |
| 100 | ob, r, done, info = self.env.step(action) |