5 files changed, 7 insertions, 6 deletions
diff --git a/text_recognizer/networks/convnext/__init__.py b/text_recognizer/networks/convnext/__init__.py
index 1743f09..faebe6f 100644
--- a/text_recognizer/networks/convnext/__init__.py
+++ b/text_recognizer/networks/convnext/__init__.py
@@ -1,7 +1,7 @@
 """Convnext module."""
-from text_recognizer.networks.convnext.convnext import ConvNext
 from text_recognizer.networks.convnext.attention import (
     Attention,
     FeedForward,
     TransformerBlock,
 )
+from text_recognizer.networks.convnext.convnext import ConvNext
diff --git a/text_recognizer/networks/convnext/attention.py b/text_recognizer/networks/convnext/attention.py
index 7f03436..a373c37 100644
--- a/text_recognizer/networks/convnext/attention.py
+++ b/text_recognizer/networks/convnext/attention.py
@@ -1,8 +1,8 @@
 """Convolution self attention block."""
 
-from einops import reduce, rearrange
-from torch import einsum, nn, Tensor
 import torch.nn.functional as F
+from einops import rearrange, reduce
+from torch import Tensor, einsum, nn
 
 from text_recognizer.networks.convnext.norm import LayerNorm
 from text_recognizer.networks.convnext.residual import Residual
diff --git a/text_recognizer/networks/convnext/downsample.py b/text_recognizer/networks/convnext/downsample.py
index 6e4306f..c28ecca 100644
--- a/text_recognizer/networks/convnext/downsample.py
+++ b/text_recognizer/networks/convnext/downsample.py
@@ -1,7 +1,7 @@
 from typing import Tuple
 
 from einops.layers.torch import Rearrange
-from torch import nn, Tensor
+from torch import Tensor, nn
 
 
 class Downsample(nn.Module):
diff --git a/text_recognizer/networks/convnext/norm.py b/text_recognizer/networks/convnext/norm.py
index 2d896e5..23cf07a 100644
--- a/text_recognizer/networks/convnext/norm.py
+++ b/text_recognizer/networks/convnext/norm.py
@@ -1,6 +1,6 @@
 """Layer norm for conv layers."""
 import torch
-from torch import nn, Tensor
+from torch import Tensor, nn
 
 
 class LayerNorm(nn.Module):
diff --git a/text_recognizer/networks/convnext/residual.py b/text_recognizer/networks/convnext/residual.py
index 3f44390..8e76ae9 100644
--- a/text_recognizer/networks/convnext/residual.py
+++ b/text_recognizer/networks/convnext/residual.py
@@ -1,6 +1,7 @@
 """Generic residual layer."""
 from typing import Callable
-from torch import nn, Tensor
+
+from torch import Tensor, nn
 
 
 class Residual(nn.Module):