Let's see if this change of CDRequest works.

MBTMBTMBT · Jul 3, 2024 · 76d5acd · 76d5acd
1 parent 82e656b
commit 76d5acd
Showing 1 changed file with 9 additions and 12 deletions.
diff --git a/skills/src/lasr_skills/vision/get_cropped_image.py b/skills/src/lasr_skills/vision/get_cropped_image.py
@@ -44,18 +44,15 @@ def __init__(
         )
 
     def execute(self, userdata) -> str:
-        cropped_detection_req: CroppedDetectionRequest = CroppedDetectionRequest(
-            [
-                CDRequest(
-                    method=self.method,
-                    use_mask=self.use_mask,
-                    yolo_model=self.yolo_model,
-                    yolo_model_confidence=self.yolo_model_confidence,
-                    yolo_nms_threshold=self.yolo_nms_threshold,
-                    object_names=[self.object_name],
-                )
-            ]
-        )
+        req = CDRequest()
+        req.method=self.method
+        req.use_mask=self.use_mask
+        req.yolo_model=self.yolo_model
+        req.yolo_model_confidence=self.yolo_model_confidence
+        req.yolo_nms_threshold=self.yolo_nms_threshold
+        req.object_names=[self.object_name]
+        cropped_detection_req: CroppedDetectionRequest = CroppedDetectionRequest()
+        cropped_detection_req.requests = [req]
 
         try:
             cropped_detection_resp: CroppedDetectionResponse = self._cropped_detection(