Merge pull request #10 from DagsHub/bug/missing-filename-in-poses

Fix missing filename in _reimport_poses

Author: guysmoilov

dagshub_annotation_converter/formats/label_studio/task.py

@@ -143,6 +143,7 @@ def _reimport_poses(self, annotations: List[IRImageAnnotationBase]) -> List[IRIm
             category: Optional[str] = None
             image_width: Optional[int] = None
             image_height: Optional[int] = None
+            filename: Optional[str] = None
             if maybe_bbox is None:
                 logger.warning(
                     f"Bounding box of pose with annotation ID {bbox_id} "
@@ -155,6 +156,7 @@ def _reimport_poses(self, annotations: List[IRImageAnnotationBase]) -> List[IRIm
                 category = bbox.ensure_has_one_category()
                 image_height = bbox.image_height
                 image_width = bbox.image_width
+                filename = bbox.filename
                 annotations_to_remove.add(bbox_id)
             # Fetch the points
             points: List[IRPosePoint] = []
@@ -176,6 +178,8 @@ def _reimport_poses(self, annotations: List[IRImageAnnotationBase]) -> List[IRIm
                         image_width = maybe_point.image_width
                     if image_height is None:
                         image_height = maybe_point.image_height
+                    if filename is None:
+                        filename = maybe_point.filename
                     points.extend(maybe_point.points)
                     annotations_to_remove.add(point_id)
 
@@ -193,6 +197,7 @@ def _reimport_poses(self, annotations: List[IRImageAnnotationBase]) -> List[IRIm
                 coordinate_style=CoordinateStyle.NORMALIZED,
                 image_width=image_width,
                 image_height=image_height,
+                filename=filename,
             )
             if bbox is not None:
                 sum_annotation.width = bbox.width