def __init__()

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

• 25 lines of code
• 2 McCabe index (conditional complexity)

    def __init__(self, max_episode_steps=500, randomize_ball=True, **kwargs):
        super(KickAndDefend, self).__init__(**kwargs)
        self._max_episode_steps = max_episode_steps
        self._elapsed_steps = 0
        self.GOAL_REWARD = 1000
        self.ball_jnt_id = self.env_scene.model.joint_names.index(six.b('ball'))
        self.jnt_nqpos = Agent.JNT_NPOS[int(self.env_scene.model.jnt_type[self.ball_jnt_id])]
        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
        # print("GOAL_X:", self.GOAL_X)
        self.GOAL_Y = 3
        self.randomize_ball = randomize_ball
        self.LIM_X = [(-3.5, -0.5), (1.6, 3.5)]
        # self.RANGE_X = self.LIM_X = [(-2, -2), (1.6, 3.5)]
        self.LIM_Y = [(-2, 2), (-2, 2)]
        self.RANGE_X = self.LIM_X.copy()
        self.RANGE_Y = self.LIM_Y.copy()
        self.BALL_LIM_X = (-2, 1)
        self.BALL_LIM_Y = (-4, 4)
        self.BALL_RANGE_X = self.BALL_LIM_X
        self.BALL_RANGE_Y = self.BALL_LIM_Y
        self.keeper_touched_ball = False