[isort] fix importing (#367)

Lightning-Universe · Nov 16, 2020 · 6409e84 · 6409e84
1 parent 981d624
commit 6409e84
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pl_bolts/models/self_supervised/swav/swav_module.py b/pl_bolts/models/self_supervised/swav/swav_module.py
@@ -17,9 +17,9 @@
 from pl_bolts.models.self_supervised.swav.swav_resnet import resnet18, resnet50
 from pl_bolts.optimizers.lars_scheduling import LARSWrapper
 from pl_bolts.transforms.dataset_normalizations import (
-    stl10_normalization,
     cifar10_normalization,
-    imagenet_normalization
+    imagenet_normalization,
+    stl10_normalization,
 )
 
 
@@ -479,8 +479,8 @@ def add_model_specific_args(parent_parser):
 
 def cli_main():
     from pl_bolts.callbacks.ssl_online import SSLOnlineEvaluator
-    from pl_bolts.models.self_supervised.swav.transforms import SwAVTrainDataTransform, SwAVEvalDataTransform
-    from pl_bolts.datamodules import STL10DataModule, CIFAR10DataModule, ImagenetDataModule
+    from pl_bolts.datamodules import CIFAR10DataModule, ImagenetDataModule, STL10DataModule
+    from pl_bolts.models.self_supervised.swav.transforms import SwAVEvalDataTransform, SwAVTrainDataTransform
 
     parser = ArgumentParser()