Fix flaky test: TestCustomRTMDetInsSepBNHead.test_mask_pred (#3333)

* add fix * fix
openvinotoolkit · Apr 17, 2024 · c8eae43 · c8eae43
1 parent 6cce900
commit c8eae43
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/tests/unit/algo/instance_segmentation/heads/test_custom_rtmdet_ins_head.py b/tests/unit/algo/instance_segmentation/heads/test_custom_rtmdet_ins_head.py
@@ -67,12 +67,6 @@ def test_mask_pred(self, mocker) -> None:
             cfg=test_cfg,
         )
 
-        mask_head._bbox_mask_post_process(
-            results[0],
-            mask_feat,
-            cfg=None,
-        )
-
     def test_predict_by_feat_ov(self, tmpdir) -> None:
         lit_module = RTMDetInst(num_classes=1, variant="tiny")
         exported_model_path = lit_module.export(