def _update_red_robot_state(self, robot: Robot, red_agent: redAgent):
red_agent.pos = robot.get_pos()
red_agent.health = robot.get_health()
red_agent.angle = robot.get_angle()
def _update_blue_robot_state(self, robot: Robot, blue_agent: blueAgent):
blue_agent.pos = robot.get_pos()
blue_agent.health = robot.get_health()
blue_agent.angle = robot.get_angle()
def _update_robot_state(self, robot: Robot, state: RobotState):
state.pos = robot.get_pos()
state.health = robot.get_health()
state.angle = robot.get_angle()
state.velocity = robot.get_velocity()
state.angular = robot.get_angular()
