def goal_pos()

in safety_gym/envs/engine.py [0:0]

• 12 lines of code
• 6 McCabe index (conditional complexity)

    def goal_pos(self):
        ''' Helper to get goal position from layout '''
        if self.task in ['goal', 'push']:
            return self.data.get_body_xpos('goal').copy()
        elif self.task == 'button':
            return self.data.get_body_xpos(f'button{self.goal_button}').copy()
        elif self.task == 'circle':
            return ORIGIN_COORDINATES
        elif self.task == 'none':
            return np.zeros(2)  # Only used for screenshots
        else:
            raise ValueError(f'Invalid task {self.task}')