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, version=None):
        self._elapsed_steps = 0
        self.keeper_touched_ball = False
        _ = self.env_scene.reset()
        if version is not None:
            self._reset_range(version)
        for i in range(self.n_agents):
            x = np.random.uniform(*self.RANGE_X[i])
            y = np.random.uniform(*self.RANGE_Y[i])
            # print("setting agent {} to pos {}".format(i, (x,y)))
            self.agents[i].set_xyz((x, y, None))
            self.agents[i].reset_agent()
        if self.randomize_ball:
            self._set_random_ball_pos()
        return self._get_obs()