Fix/last pr (#659)

bit-bots · Jan 22, 2025 · 9ccf0e4 · 9ccf0e4
2 parents 29f0423 + 19beac3
commit 9ccf0e4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/bitbots_vision/bitbots_vision/vision_modules/yoeo/utils.py b/bitbots_vision/bitbots_vision/vision_modules/yoeo/utils.py
@@ -5,7 +5,7 @@
 import cv2
 import numpy as np
 import rclpy
-from jaxtyping import UInt8
+from jaxtyping import Float64, UInt8
 
 logger = rclpy.logging.get_logger("yoeo_handler_utils")
 
@@ -158,10 +158,10 @@ def get_info(self) -> ImagePreProcessorData:
             padding_bottom=self._padding_bottom,
             padding_left=self._padding_left,
             padding_right=self._padding_right,
-            max_dim=np.max(self._image_dimensions_HW),
+            max_dim=int(np.max(self._image_dimensions_HW)),
         )
 
-    def process(self, image: UInt8[np.ndarray, "h w 3"]) -> UInt8[np.ndarray, "3 network_input_h network_input_w"]:
+    def process(self, image: UInt8[np.ndarray, "h w 3"]) -> Float64[np.ndarray, "3 network_input_h network_input_w"]:
         self._image_dimensions_HW = image.shape[:2]  # type: ignore[assignment]
         self._calculate_paddings()
 

diff --git a/sync_includes_wolfgang_nuc.yaml b/sync_includes_wolfgang_nuc.yaml
@@ -15,7 +15,6 @@ include:
         - bitbots_parameter_blackboard
         - bitbots_robot_description
         - bitbots_teleop
-        - bitbots_tf_buffer
         - bitbots_tts
         - bitbots_utils
         - system_monitor
@@ -51,6 +50,7 @@ include:
         - bio_ik
         - bio_ik_msgs
         - biped_interfaces
+        - bitbots_tf_buffer
         - dynamic_stack_decider
         - dynamixel-workbench
         - DynamixelSDK