diff --git a/train.py b/train.py
index d9246b6..d64a7f2 100644
--- a/train.py
+++ b/train.py
@@ -1,7 +1,8 @@
-import pytorch_lightning
-import torch
 from argparse import ArgumentParser
+
 import omegaconf
+import pytorch_lightning
+import torch
 import torchvision