Fix flake8

BloodAxe · Nov 21, 2024 · e4caad5 · e4caad5
1 parent 66e48d7
commit e4caad5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/pytorch_toolbelt/losses/focal.py b/pytorch_toolbelt/losses/focal.py
@@ -71,7 +71,7 @@ def __repr__(self):
         repr += f"reduced_threshold={self.reduced_threshold}, activation={self.activation}, "
         repr += f"softmax_dim={self.softmax_dim},"
         repr += f"class_weights={self.class_weights.tolist()}, "
-        repr += f")"
+        repr += ")"
         return repr
 
     def forward(self, inputs: Tensor, targets: Tensor) -> Tensor:

diff --git a/pytorch_toolbelt/modules/interfaces.py b/pytorch_toolbelt/modules/interfaces.py
@@ -61,7 +61,8 @@ class HasInputFeaturesSpecification(Protocol):
     """
 
     @torch.jit.unused
-    def get_input_spec(self) -> FeatureMapsSpecification: ...
+    def get_input_spec(self) -> FeatureMapsSpecification:
+        ...
 
 
 class HasOutputFeaturesSpecification(Protocol):
@@ -70,7 +71,8 @@ class HasOutputFeaturesSpecification(Protocol):
     """
 
     @torch.jit.unused
-    def get_output_spec(self) -> FeatureMapsSpecification: ...
+    def get_output_spec(self) -> FeatureMapsSpecification:
+        ...
 
 
 class AbstractEncoder(nn.Module, HasOutputFeaturesSpecification):