Black reformat

BloodAxe · Aug 19, 2023 · dd8fc7f · dd8fc7f
1 parent 79bb9ba
commit dd8fc7f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pytorch_toolbelt/modules/heads/deep_supervision.py b/pytorch_toolbelt/modules/heads/deep_supervision.py
@@ -51,4 +51,4 @@ def get_output_spec(self) -> FeatureMapsSpecification:
 
     def apply_to_final_layer(self, func: Callable[[nn.Module], None]):
         for head in self.heads:
-            func(head)
+            func(head)
diff --git a/pytorch_toolbelt/modules/heads/progressive_shuffle.py b/pytorch_toolbelt/modules/heads/progressive_shuffle.py
@@ -81,4 +81,4 @@ def forward(self, feature_maps, output_size=None):
             return output
 
     def apply_to_final_layer(self, func: Callable[[nn.Module], None]):
-        func(self.final)
+        func(self.final)