1 files changed, 1 insertions, 5 deletions
diff --git a/text_recognizer/networks/transformer/layers.py b/text_recognizer/networks/transformer/layers.py
index a44a525..1266d32 100644
--- a/text_recognizer/networks/transformer/layers.py
+++ b/text_recognizer/networks/transformer/layers.py
@@ -47,11 +47,7 @@ class AttentionLayers(nn.Module):
         return "a", "f"
 
     def _build_network(
-        self,
-        causal: bool,
-        attn_fn: partial,
-        norm_fn: partial,
-        ff_fn: partial,
+        self, causal: bool, attn_fn: partial, norm_fn: partial, ff_fn: partial,
     ) -> nn.ModuleList:
         """Configures transformer network."""
         layers = nn.ModuleList([])