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

for more information, see https://pre-commit.ci
Lightning-AI · Jan 8, 2022 · 7a7fd7b · 7a7fd7b
1 parent df6f123
commit 7a7fd7b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/torchmetrics/__init__.py b/torchmetrics/__init__.py
@@ -13,7 +13,15 @@
 
 from torchmetrics import functional  # noqa: E402
 from torchmetrics.aggregation import CatMetric, MaxMetric, MeanMetric, MinMetric, SumMetric  # noqa: E402
-from torchmetrics.audio import PIT, SDR, SI_SDR, SI_SNR, SNR, ScaleInvariantSignalDistortionRatio, SignalDistortionRatio  # noqa: E402
+from torchmetrics.audio import (  # noqa: E402
+    PIT,
+    SDR,
+    SI_SDR,
+    SI_SNR,
+    SNR,
+    ScaleInvariantSignalDistortionRatio,
+    SignalDistortionRatio,
+)
 from torchmetrics.classification import (  # noqa: E402, F401
     AUC,
     AUROC,

diff --git a/torchmetrics/audio/si_sdr.py b/torchmetrics/audio/si_sdr.py
@@ -40,5 +40,8 @@ def __init__(
         process_group: Optional[Any] = None,
         dist_sync_fn: Optional[Callable[[Tensor], Tensor]] = None,
     ) -> None:
-        warn("`SI_SDR` was renamed to `ScaleInvariantSignalDistortionRatio` in v0.7 and it will be removed in v0.8", DeprecationWarning)
+        warn(
+            "`SI_SDR` was renamed to `ScaleInvariantSignalDistortionRatio` in v0.7 and it will be removed in v0.8",
+            DeprecationWarning,
+        )
         super().__init__(zero_mean, compute_on_step, dist_sync_on_step, process_group, dist_sync_fn)