(self, frame, action, reward, done)
| 70 | self.size = 0 |
| 71 | |
| 72 | def push(self, frame, action, reward, done): |
| 73 | self.frames[self.idx] = frame |
| 74 | self.actions[self.idx] = action |
| 75 | self.rewards[self.idx] = reward |
| 76 | self.dones[self.idx] = float(done) |
| 77 | self.idx = (self.idx + 1) % self.capacity |
| 78 | self.size = min(self.size + 1, self.capacity) |
| 79 | |
| 80 | def _stack(self, idx): |
| 81 | # Gather frames[idx-stack+1 .. idx]; newest at last channel. |