8c17debc9dc6552da8c317c0c852b55f03c80c61,slm_lab/env/__init__.py,OpenAIEnv,step,#OpenAIEnv#Any#,112

Before Change

        if not self.train_mode:
            self.u_env.render()
        body_action = action[0]
        body_state, body_reward, body_done, _info = self.u_env.step(
            body_action)
        reward = [body_reward]
        state = [body_state]

After Change

        assert len(action_e) == 1, "OpenAI Gym supports only single body"
        action = action_e[(0, 0)]
        (state, reward, done, _info) = self.u_env.step(action)
        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):
            reward_e[(a, b)] = reward
            state_e[(a, b)] = state

Instances