From ecf050a59ac42c9cf7d12dab44dd51a341c916f2 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 7 Jun 2021 22:59:57 +0200 Subject: Removed docs pipeline in noxfile, reformatted code w black --- text_recognizer/networks/transformer/layers.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) (limited to 'text_recognizer/networks/transformer') 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([]) -- cgit v1.2.3-70-g09d2