(self)
| 82 | return np.stack(obs), np.stack(rews), np.stack(dones), infos |
| 83 | |
| 84 | def reset(self): |
| 85 | for remote in self.remotes: |
| 86 | remote.send(('reset', None)) |
| 87 | return np.stack([remote.recv() for remote in self.remotes]) |
| 88 | |
| 89 | def reset_task(self): |
| 90 | for remote in self.remotes: |