8c17debc9dc6552da8c317c0c852b55f03c80c61,slm_lab/env/__init__.py,Env,step,#Env#Any#,202

Before Change

    def step(self, action):
        env_info_dict = self.u_env.step(action)
        reward = []
        state = []
        done = []
        for a, b in self.ab_proj:
            a_name = self.u_env.brain_names[a]
            a_env_info = env_info_dict[a_name]

After Change

        reward_e = self.data_spaces["reward"].init_data_s(e=self.e)
        state_e = self.data_spaces["state"].init_data_s(e=self.e)
        done_e = self.data_spaces["done"].init_data_s(e=self.e)
        for (a, b), body in util.ndenumerate_nonan(self.body_e):
            env_info_a = self.get_env_info(env_info_dict, a)
            reward_e[(a, b)] = env_info_a.rewards[b]
            state_e[(a, b)] = env_info_a.states[b]
            done_e[(a, b)] = env_info_a.local_done[b]
        return reward_e, state_e, done_e

    def close(self):

Instances