def reset()

in gym-compete/gym_compete/new_envs/kick_and_defend.py [0:0]

• 14 lines of code
• 4 McCabe index (conditional complexity)

    def reset(self, margins=None, version=None):
        _ = self._reset(version=version)
        if self.agents[0].team == 'walker':
            self.walker_id = 0
            self.blocker_id = 1
        else:
            self.walker_id = 1
            self.blocker_id = 0
        self.GOAL_X = self.agents[self.walker_id].TARGET
        if margins is not None:
            for i in range(self.n_agents):
                self.agents[i].set_margin(margins[i])
        # print("GOAL_X:", self.GOAL_X)
        ob = self._get_obs()
        return ob