diff --git a/bitbots_vision/bitbots_vision/vision_modules/yoeo/vision_components.py b/bitbots_vision/bitbots_vision/vision_modules/yoeo/vision_components.py index b668c5a3..0495117f 100644 --- a/bitbots_vision/bitbots_vision/vision_modules/yoeo/vision_components.py +++ b/bitbots_vision/bitbots_vision/vision_modules/yoeo/vision_components.py @@ -8,7 +8,8 @@ from sensor_msgs.msg import Image from typing import Dict, Optional, List, Tuple -from bitbots_msgs.msg import Audio, GameState +from bitbots_msgs.msg import Audio +from game_controller_hl_interfaces.msg import GameState from soccer_vision_2d_msgs.msg import BallArray, GoalpostArray, RobotArray, Robot from bitbots_vision.vision_modules import debug, ros_utils, candidate diff --git a/bitbots_vision/package.xml b/bitbots_vision/package.xml index d2f24f63..df54a776 100644 --- a/bitbots_vision/package.xml +++ b/bitbots_vision/package.xml @@ -28,6 +28,7 @@ rosidl_default_runtime bitbots_msgs bitbots_utils + game_controller_hl_interfaces geometry_msgs image_transport python3-numpy