diff --git a/perception_eval/perception_eval/common/label.py b/perception_eval/perception_eval/common/label.py index 2bf27ff1..49a1f4ef 100644 --- a/perception_eval/perception_eval/common/label.py +++ b/perception_eval/perception_eval/common/label.py @@ -381,6 +381,7 @@ def _get_traffic_light_paris( ] else: pair_list: List[Tuple[TrafficLightLabel, str]] = [ + (TrafficLightLabel.TRAFFIC_LIGHT, "traffic_light"), (TrafficLightLabel.TRAFFIC_LIGHT, "green"), (TrafficLightLabel.TRAFFIC_LIGHT, "green_straight"), (TrafficLightLabel.TRAFFIC_LIGHT, "green_left"), diff --git a/perception_eval/test/common/test_label.py b/perception_eval/test/common/test_label.py index 3ea259b1..d7fd633f 100644 --- a/perception_eval/test/common/test_label.py +++ b/perception_eval/test/common/test_label.py @@ -65,10 +65,10 @@ (TrafficLightLabel.YELLOW, "yellow"), (TrafficLightLabel.RED_STRAIGHT, "red_straight"), (TrafficLightLabel.RED_LEFT, "red_left"), - (TrafficLightLabel.RED_STRAIGHT_LEFT, "red_left_straight"), + (TrafficLightLabel.RED_STRAIGHT_LEFT, "red_straight_left"), (TrafficLightLabel.RED_LEFT_DIAGONAL, "red_left_diagonal"), (TrafficLightLabel.RED_RIGHT, "red_right"), - (TrafficLightLabel.RED_STRAIGHT_RIGHT, "red_right_straight"), + (TrafficLightLabel.RED_STRAIGHT_RIGHT, "red_straight_right"), (TrafficLightLabel.RED_RIGHT_DIAGONAL, "red_right_diagonal"), (TrafficLightLabel.YELLOW_RIGHT, "yellow_right"), (TrafficLightLabel.UNKNOWN, "unknown"),