linter loss.py

deel-ai · Oct 9, 2024 · f4c38c2 · f4c38c2
1 parent fbedaac
commit f4c38c2
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/deel/torchlip/modules/loss.py b/deel/torchlip/modules/loss.py
@@ -24,7 +24,6 @@
 # rights reserved. DEEL is a research program operated by IVADO, IRT Saint Exupéry,
 # CRIAQ and ANITI - https://www.deel.ai/
 # =====================================================================================
-from typing import Tuple
 import warnings
 import torch
 from .. import functional as F
@@ -110,7 +109,8 @@ def __init__(
             self.alpha = alpha
         else:
             warnings.warn(
-                f"Depreciated alpha should be between 0 and 1 replaced by {alpha/(alpha+1.0)}"
+                f"Depreciated alpha should be between 0 and 1 replaced by \
+                    {alpha/(alpha+1.0)}"
             )
             self.alpha = alpha / (alpha + 1.0)
         self.min_margin = min_margin
@@ -171,7 +171,8 @@ def __init__(
             self.alpha = alpha
         else:
             warnings.warn(
-                f"Depreciated alpha should be between 0 and 1 replaced by {alpha/(alpha+1.0)}"
+                f"Depreciated alpha should be between 0 and 1 replaced by \
+                    {alpha/(alpha+1.0)}"
             )
             self.alpha = alpha / (alpha + 1.0)
         self.min_margin = min_margin
@@ -212,7 +213,8 @@ def __init__(
             self.alpha = torch.tensor(alpha, dtype=torch.float32)
         else:
             warnings.warn(
-                f"Depreciated alpha should be between 0 and 1 replaced by {alpha/(alpha+1.0)}"
+                f"Depreciated alpha should be between 0 and 1 replaced by \
+                    {alpha/(alpha+1.0)}"
             )
             self.alpha = torch.tensor(alpha / (alpha + 1.0), dtype=torch.float32)
         self.min_margin_v = min_margin