Fix mypy error

VectorInstitute · Jan 12, 2024 · 4fd59d1 · 4fd59d1
1 parent 244d164
commit 4fd59d1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cyclops/evaluate/metrics/experimental/auroc.py b/cyclops/evaluate/metrics/experimental/auroc.py
@@ -69,7 +69,7 @@ def __init__(
         )
         self.max_fpr = max_fpr
 
-    def _compute_metric(self) -> Array:
+    def _compute_metric(self) -> Array:  # type: ignore[override]
         """Compute the AUROC.""" ""
         state = (
             (dim_zero_cat(self.target), dim_zero_cat(self.preds))  # type: ignore[attr-defined]
@@ -155,7 +155,7 @@ def __init__(
         )
         self.average = average  # type: ignore[assignment]
 
-    def _compute_metric(self) -> Array:
+    def _compute_metric(self) -> Array:  # type: ignore[override]
         """Compute the AUROC."""
         state = (
             (dim_zero_cat(self.target), dim_zero_cat(self.preds))  # type: ignore[attr-defined]
@@ -242,7 +242,7 @@ def __init__(
         )
         self.average = average
 
-    def _compute_metric(self) -> Array:
+    def _compute_metric(self) -> Array:  # type: ignore[override]
         """Compute the AUROC."""
         state = (
             (dim_zero_cat(self.target), dim_zero_cat(self.preds))  # type: ignore[attr-defined]