fmt

Lightning-AI · Jan 14, 2022 · ee0ecce · ee0ecce
1 parent cfee310
commit ee0ecce
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 13 deletions.
diff --git a/torchmetrics/detection/__init__.py b/torchmetrics/detection/__init__.py
@@ -15,4 +15,4 @@
 from torchmetrics.utilities.imports import _TORCHVISION_AVAILABLE, _TORCHVISION_GREATER_EQUAL_0_8
 
 if _TORCHVISION_AVAILABLE and _TORCHVISION_GREATER_EQUAL_0_8:
- from torchmetrics.detection.map import MeanAveragePrecision
+ from torchmetrics.detection.map import MeanAveragePrecision # noqa: F401
diff --git a/torchmetrics/detection/map.py b/torchmetrics/detection/map.py
@@ -432,7 +432,9 @@ def _evaluate_image(
  if torch.numel(ious) > 0:
  for idx_iou, t in enumerate(self.iou_thresholds):
  for idx_det in range(nb_det):
- m = MeanAveragePrecision._find_best_gt_match(t, nb_gt, gt_matches, idx_iou, gt_ignore, ious, idx_det)
+ m = MeanAveragePrecision._find_best_gt_match(
+ t, nb_gt, gt_matches, idx_iou, gt_ignore, ious, idx_det
+ )
  if m != -1:
  det_ignore[idx_iou, idx_det] = gt_ignore[m]
  det_matches[idx_iou, idx_det] = True
@@ -780,9 +782,7 @@ class MAP(MeanAveragePrecision):
  }
  """
 
- @deprecated(
- target=MeanAveragePrecision, deprecated_in="0.7", remove_in="0.8", stream=_future_warning
- )
+ @deprecated(target=MeanAveragePrecision, deprecated_in="0.7", remove_in="0.8", stream=_future_warning)
  def __init__(
  self,
  box_format: str = "xyxy",
@@ -797,12 +797,12 @@ def __init__(
  ) -> None: # type: ignore
  void(
  box_format,
-  iou_thresholds,
-  rec_thresholds,
-  max_detection_thresholds,
-  class_metrics,
-  compute_on_step,
-  dist_sync_on_step,
-  process_group,
-  dist_sync_fn
+ iou_thresholds,
+ rec_thresholds,
+ max_detection_thresholds,
+ class_metrics,
+ compute_on_step,
+ dist_sync_on_step,
+ process_group,
+ dist_sync_fn,
  )