Catch warnings in detection metric (#1451)

deepchecks · May 16, 2022 · 747a4c0 · 747a4c0
1 parent 5cc5db6
commit 747a4c0
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/deepchecks/vision/metrics_utils/detection_precision_recall.py b/deepchecks/vision/metrics_utils/detection_precision_recall.py
@@ -329,11 +329,11 @@ def get_classes_scores_at(self, res: np.ndarray, iou: float = None, area: str =
         with warnings.catch_warnings():
             warnings.simplefilter(action="ignore", category=RuntimeWarning)
             res = np.nanmean(res[:, :, :], axis=0)
-        if get_mean_val:
-            return np.nanmean(res[res > -1])
-        if zeroed_negative:
-            res = res.clip(min=0)
-        return res[0][0]
+            if get_mean_val:
+                return np.nanmean(res[res > -1])
+            if zeroed_negative:
+                res = res.clip(min=0)
+            return res[0][0]
 
     @abstractmethod
     def get_confidences(self, detections) -> List[float]: