Apply isort

Lightning-Universe · Dec 24, 2020 · 158503c · 158503c
1 parent 59ce57a
commit 158503c
Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/pl_bolts/models/self_supervised/amdim/__init__.py b/pl_bolts/models/self_supervised/amdim/__init__.py
@@ -9,7 +9,6 @@
     AMDIMTrainTransformsSTL10,
 )
 
-
 __all__ = [
     "AMDIM",
     "AMDIMEncoder",

diff --git a/pl_bolts/models/self_supervised/amdim/amdim_module.py b/pl_bolts/models/self_supervised/amdim/amdim_module.py
@@ -7,8 +7,8 @@
 import torch.optim as optim
 from torch.utils.data import DataLoader
 
-from pl_bolts.models.self_supervised.amdim.datasets import AMDIMPretraining
 from pl_bolts.losses.self_supervised_learning import FeatureMapContrastiveTask
+from pl_bolts.models.self_supervised.amdim.datasets import AMDIMPretraining
 from pl_bolts.models.self_supervised.amdim.networks import AMDIMEncoder
 from pl_bolts.utils.self_supervised import torchvision_ssl_encoder
 

diff --git a/pl_bolts/models/self_supervised/amdim/datasets.py b/pl_bolts/models/self_supervised/amdim/datasets.py
@@ -2,11 +2,11 @@
 
 from torch.utils.data import random_split
 
-from pl_bolts.utils.warnings import warn_missing_pkg
 from pl_bolts.datasets.imagenet_dataset import UnlabeledImagenet
 from pl_bolts.datasets.ssl_amdim_datasets import CIFAR10Mixed
 from pl_bolts.models.self_supervised.amdim import transforms as amdim_transforms
 from pl_bolts.utils import _TORCHVISION_AVAILABLE
+from pl_bolts.utils.warnings import warn_missing_pkg
 
 if _TORCHVISION_AVAILABLE:
     from torchvision.datasets import STL10