Update todo

openvinotoolkit · Dec 13, 2024 · 299bca1 · 299bca1
1 parent dab90d8
commit 299bca1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/otx/core/data/dataset/instance_segmentation.py b/src/otx/core/data/dataset/instance_segmentation.py
@@ -50,7 +50,7 @@ def _get_item_impl(self, index: int) -> InstanceSegDataEntity | None:
 
         gt_bboxes, gt_labels, gt_masks, gt_polygons = [], [], [], []
 
-        # NOTE (Eugene):
+        # TODO(Eugene): https://jira.devtools.intel.com/browse/CVS-159363
         # Temporary solution to handle multiple annotation types.
         # Ideally, we should pre-filter annotations during initialization of the dataset.
         if Polygon.__name__ in anno_collection:  # Polygon for InstSeg has higher priority

diff --git a/src/otx/core/data/dataset/tile.py b/src/otx/core/data/dataset/tile.py
@@ -514,9 +514,10 @@ def _get_item_impl(self, index: int) -> TileInstSegDataEntity:  # type: ignore[o
 
         gt_bboxes, gt_labels, gt_masks, gt_polygons = [], [], [], []
 
-        # NOTE (Eugene):
+        # TODO(Eugene): https://jira.devtools.intel.com/browse/CVS-159363
         # Temporary solution to handle multiple annotation types.
         # Ideally, we should pre-filter annotations during initialization of the dataset.
+
         if Polygon.__name__ in anno_collection:  # Polygon for InstSeg has higher priority
             for poly in anno_collection[Polygon.__name__]:
                 bbox = Bbox(*poly.get_bbox()).points