[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
Lightning-AI · Jan 9, 2022 · 17fa7ec · 17fa7ec
1 parent 3a7f051
commit 17fa7ec
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/tests/classification/test_f_beta.py b/tests/classification/test_f_beta.py
@@ -33,7 +33,8 @@
 from tests.helpers import seed_all
 from tests.helpers.testers import NUM_BATCHES, NUM_CLASSES, THRESHOLD, MetricTester
 from torchmetrics import F1Score, FBeta, Metric
-from torchmetrics.functional import f1_score as f1_score_pl, fbeta
+from torchmetrics.functional import f1_score as f1_score_pl
+from torchmetrics.functional import fbeta
 from torchmetrics.utilities.checks import _input_format_classification
 from torchmetrics.utilities.enums import AverageMethod
 
@@ -183,7 +184,7 @@ def test_no_support(metric_class, metric_fn):
     [
         (partial(FBeta, beta=2.0), partial(fbeta, beta=2.0)),
         (F1Score, f1_score_pl),
-    ]
+    ],
 )
 @pytest.mark.parametrize(
     "ignore_index, expected", [(None, torch.tensor([1.0, np.nan])), (0, torch.tensor([np.nan, np.nan]))]
@@ -441,7 +442,7 @@ def test_top_k(
     "metric_class, metric_functional, sk_fn",
     [
         (partial(FBeta, beta=2.0), partial(fbeta, beta=2.0), partial(fbeta_score, beta=2.0)),
-        (F1Score, f1_score_pl, f1_score)
+        (F1Score, f1_score_pl, f1_score),
     ],
 )
 def test_same_input(metric_class, metric_functional, sk_fn, average, ignore_index):

diff --git a/torchmetrics/__init__.py b/torchmetrics/__init__.py
@@ -27,7 +27,6 @@
 from torchmetrics.classification import (  # noqa: E402, F401
     AUC,
     AUROC,
-    F1Score,
     F1,
     ROC,
     Accuracy,
@@ -38,6 +37,7 @@
     CalibrationError,
     CohenKappa,
     ConfusionMatrix,
+    F1Score,
     FBeta,
     HammingDistance,
     Hinge,

diff --git a/torchmetrics/functional/classification/f_beta.py b/torchmetrics/functional/classification/f_beta.py
@@ -350,6 +350,7 @@ def f1_score(
     """
     return fbeta(preds, target, 1.0, average, mdmc_average, ignore_index, num_classes, threshold, top_k, multiclass)
 
+
 def f1(
     preds: Tensor,
     target: Tensor,
-Original file line number
+Diff line change
@@ Expand Up / @@ -350,6 +350,7 @@ def f1_score( @@
         """
         return fbeta(preds, target, 1.0, average, mdmc_average, ignore_index, num_classes, threshold, top_k, multiclass)
     def f1(
         preds: Tensor,
         target: Tensor,
@@ Expand Down @@