c60fbc4da1b0b1732d8a4e15f6a60c48251150b4,torch_rl/torch_rl/utils/penv.py,ParallelEnv,step,#ParallelEnv#Any#,49

Before Change

        if self.unique_env:
            obs, reward, done, info = self.env.step(actions[0])
            if done:
                obs = self.env.reset()
            return zip(*[(obs, reward, done, info)])
        else:
            for local, action in zip(self.locals, actions):
                local.send(("step", action))

After Change

            local.send(("step", action))
        obs, reward, done, info = self.envs[0].step(actions[0])
        if done:
            obs = self.envs[0].reset()
        results = zip(*[(obs, reward, done, info)] + [local.recv() for local in self.locals])
        return results

    def render(self):
        raise NotImplementedError

Instances