Apply isort

pl_bolts/callbacks/data_monitor.py

@@ -1,13 +1,13 @@
 from typing import Any, Dict, List, Optional, Sequence, Union
 
 import numpy as np
+import torch
+import torch.nn as nn
 from pytorch_lightning import Callback, LightningModule, Trainer
 from pytorch_lightning.loggers import TensorBoardLogger, WandbLogger
 from pytorch_lightning.utilities import rank_zero_warn
 from pytorch_lightning.utilities.apply_func import apply_to_collection
-import torch
 from torch import Tensor
-import torch.nn as nn
 from torch.utils.hooks import RemovableHandle
 
 try:

pl_bolts/callbacks/ssl_online.py

@@ -1,8 +1,8 @@
 from typing import Optional
 
+import torch
 from pytorch_lightning import Callback
 from pytorch_lightning.metrics.functional import accuracy
-import torch
 from torch.nn import functional as F
 
 

pl_bolts/callbacks/variational.py

@@ -1,8 +1,8 @@
 import math
 
 import numpy as np
-from pytorch_lightning.callbacks import Callback
 import torch
+from pytorch_lightning.callbacks import Callback
 
 from pl_bolts.utils.warnings import warn_missing_pkg
 

pl_bolts/callbacks/vision/confused_logit.py

@@ -1,7 +1,7 @@
 import importlib
 
-from pytorch_lightning import Callback
 import torch
+from pytorch_lightning import Callback
 from torch import nn
 
 from pl_bolts.utils.warnings import warn_missing_pkg

pl_bolts/callbacks/vision/image_generation.py

@@ -1,5 +1,5 @@
-from pytorch_lightning import Callback
 import torch
+from pytorch_lightning import Callback
 
 from pl_bolts.utils.warnings import warn_missing_pkg
 

pl_bolts/datamodules/async_dataloader.py

@@ -1,5 +1,5 @@
-from queue import Queue
 import re
+from queue import Queue
 from threading import Thread
 
 import torch

pl_bolts/datamodules/kitti_datamodule.py

@@ -1,7 +1,7 @@
 import os
 
-from pytorch_lightning import LightningDataModule
 import torch
+from pytorch_lightning import LightningDataModule
 from torch.utils.data import DataLoader
 from torch.utils.data.dataset import random_split
 

pl_bolts/datamodules/sklearn_datamodule.py

@@ -2,8 +2,8 @@
 from typing import Any
 
 import numpy as np
-from pytorch_lightning import LightningDataModule
 import torch
+from pytorch_lightning import LightningDataModule
 from torch.utils.data import DataLoader, Dataset
 
 from pl_bolts.utils import _SKLEARN_AVAILABLE

pl_bolts/datamodules/stl10_datamodule.py

@@ -1,8 +1,8 @@
 import os
 from typing import Optional
 
-from pytorch_lightning import LightningDataModule
 import torch
+from pytorch_lightning import LightningDataModule
 from torch.utils.data import DataLoader, random_split
 
 from pl_bolts.datasets.concat_dataset import ConcatDataset

pl_bolts/datamodules/vision_datamodule.py

@@ -1,9 +1,9 @@
-from abc import abstractmethod
 import os
+from abc import abstractmethod
 from typing import Any, List, Optional, Union
 
-from pytorch_lightning import LightningDataModule
 import torch
+from pytorch_lightning import LightningDataModule
 from torch.utils.data import DataLoader, Dataset, random_split
 
 

pl_bolts/datamodules/vocdetection_datamodule.py

@@ -1,13 +1,13 @@
-from pytorch_lightning import LightningDataModule
 import torch
+from pytorch_lightning import LightningDataModule
 from torch.utils.data import DataLoader
 
 from pl_bolts.utils import _TORCHVISION_AVAILABLE
 from pl_bolts.utils.warnings import warn_missing_pkg
 
 if _TORCHVISION_AVAILABLE:
- from torchvision.datasets import VOCDetection
  import torchvision.transforms as T
+ from torchvision.datasets import VOCDetection
 else:
  warn_missing_pkg('torchvision') # pragma: no-cover
 

pl_bolts/datasets/base_dataset.py

@@ -1,9 +1,9 @@
-from abc import ABC
 import logging
 import os
+import urllib.request
+from abc import ABC
 from typing import Sequence, Tuple
 from urllib.error import HTTPError
-import urllib.request
 
 import torch
 from torch import Tensor

pl_bolts/datasets/imagenet_dataset.py

@@ -1,11 +1,11 @@
-from contextlib import contextmanager
 import gzip
 import hashlib
 import os
 import shutil
 import tarfile
 import tempfile
 import zipfile
+from contextlib import contextmanager
 
 import numpy as np
 import torch

pl_bolts/models/autoencoders/components.py

@@ -1,6 +1,6 @@
 import torch
-from torch import nn
 import torch.nn.functional as F
+from torch import nn
 
 
 class Interpolate(nn.Module):

pl_bolts/models/gans/basic/components.py

@@ -1,7 +1,7 @@
 import numpy as np
 import torch
-from torch import nn
 import torch.nn.functional as F
+from torch import nn
 
 
 class Generator(nn.Module):

pl_bolts/models/mnist_module.py

@@ -1,7 +1,7 @@
 from argparse import ArgumentParser
 
-from pytorch_lightning import LightningModule, Trainer
 import torch
+from pytorch_lightning import LightningModule, Trainer
 from torch.nn import functional as F
 from torch.utils.data import DataLoader, random_split
 

pl_bolts/models/regression/logistic_regression.py

@@ -1,8 +1,8 @@
 from argparse import ArgumentParser
 
 import pytorch_lightning as pl
-from pytorch_lightning.metrics.functional import accuracy
 import torch
+from pytorch_lightning.metrics.functional import accuracy
 from torch import nn
 from torch.nn import functional as F
 from torch.optim import Adam

pl_bolts/models/rl/common/agents.py

@@ -7,8 +7,8 @@
 
 import numpy as np
 import torch
-from torch import nn
 import torch.nn.functional as F
+from torch import nn
 
 
 class Agent(ABC):

pl_bolts/models/rl/common/gym_wrappers.py

@@ -11,9 +11,9 @@
 from pl_bolts.utils.warnings import warn_missing_pkg
 
 if _GYM_AVAILABLE:
+ import gym.spaces
  from gym import make as gym_make
  from gym import ObservationWrapper, Wrapper
- import gym.spaces
 else: # pragma: no-cover
  warn_missing_pkg('gym')
  Wrapper = object

pl_bolts/models/rl/dqn_model.py

@@ -8,10 +8,10 @@
 
 import numpy as np
 import pytorch_lightning as pl
-from pytorch_lightning import seed_everything
-from pytorch_lightning.callbacks import ModelCheckpoint
 import torch
 import torch.optim as optim
+from pytorch_lightning import seed_everything
+from pytorch_lightning.callbacks import ModelCheckpoint
 from torch.optim.optimizer import Optimizer
 from torch.utils.data import DataLoader
 

pl_bolts/models/rl/reinforce_model.py

@@ -4,11 +4,11 @@
 
 import numpy as np
 import pytorch_lightning as pl
+import torch
+import torch.optim as optim
 from pytorch_lightning import seed_everything
 from pytorch_lightning.callbacks import ModelCheckpoint
-import torch
 from torch.nn.functional import log_softmax
-import torch.optim as optim
 from torch.optim.optimizer import Optimizer
 from torch.utils.data import DataLoader
 

pl_bolts/models/rl/vanilla_policy_gradient_model.py

@@ -4,11 +4,11 @@
 
 import numpy as np
 import pytorch_lightning as pl
+import torch
+import torch.optim as optim
 from pytorch_lightning import seed_everything
 from pytorch_lightning.callbacks import ModelCheckpoint
-import torch
 from torch.nn.functional import log_softmax, softmax
-import torch.optim as optim
 from torch.optim.optimizer import Optimizer
 from torch.utils.data import DataLoader
 

pl_bolts/models/self_supervised/amdim/amdim_module.py

@@ -1,5 +1,5 @@
-from argparse import ArgumentParser
 import os
+from argparse import ArgumentParser
 from typing import Union
 
 import pytorch_lightning as pl

pl_bolts/models/self_supervised/amdim/networks.py

@@ -2,8 +2,8 @@
 
 import numpy as np
 import torch
-from torch import nn
 import torch.nn.functional as F
+from torch import nn
 
 
 class AMDIMEncoder(nn.Module):

pl_bolts/models/self_supervised/byol/byol_module.py

@@ -3,9 +3,9 @@
 from typing import Any
 
 import pytorch_lightning as pl
-from pytorch_lightning import seed_everything
 import torch
 import torch.nn.functional as F
+from pytorch_lightning import seed_everything
 from torch.optim import Adam
 
 from pl_bolts.callbacks.byol_updates import BYOLMAWeightUpdate

pl_bolts/models/self_supervised/cpc/cpc_finetuner.py

@@ -1,5 +1,5 @@
-from argparse import ArgumentParser
 import os
+from argparse import ArgumentParser
 
 import pytorch_lightning as pl
 

pl_bolts/models/self_supervised/cpc/cpc_module.py

@@ -2,14 +2,14 @@
 CPC V2
 ======
 """
-from argparse import ArgumentParser
 import math
+from argparse import ArgumentParser
 from typing import Optional
 
 import pytorch_lightning as pl
-from pytorch_lightning.utilities import rank_zero_warn
 import torch
 import torch.optim as optim
+from pytorch_lightning.utilities import rank_zero_warn
 
 from pl_bolts.losses.self_supervised_learning import CPCTask
 from pl_bolts.models.self_supervised.cpc.networks import cpc_resnet101

pl_bolts/models/self_supervised/cpc/networks.py

@@ -1,5 +1,5 @@
-from torch import nn
 import torch.nn.functional as F
+from torch import nn
 
 
 class CPCResNet(nn.Module):

pl_bolts/models/self_supervised/moco/moco2_module.py

@@ -15,8 +15,8 @@
 
 import pytorch_lightning as pl
 import torch
-from torch import nn
 import torch.nn.functional as F
+from torch import nn
 
 from pl_bolts.utils.warnings import warn_missing_pkg
 

pl_bolts/models/self_supervised/simclr/simclr_finetuner.py

@@ -1,5 +1,5 @@
-from argparse import ArgumentParser
 import os
+from argparse import ArgumentParser
 
 import pytorch_lightning as pl
 

pl_bolts/models/self_supervised/simclr/simclr_module.py

@@ -1,13 +1,13 @@
-from argparse import ArgumentParser
 import math
+from argparse import ArgumentParser
 from typing import Callable, Optional
 
 import numpy as np
 import pytorch_lightning as pl
-from pytorch_lightning.utilities import AMPType
 import torch
-from torch import nn
 import torch.nn.functional as F
+from pytorch_lightning.utilities import AMPType
+from torch import nn
 from torch.optim.optimizer import Optimizer
 
 from pl_bolts.models.self_supervised.resnets import resnet18, resnet50

pl_bolts/models/self_supervised/ssl_finetuner.py

@@ -1,9 +1,9 @@
 from typing import List, Optional
 
 import pytorch_lightning as pl
-from pytorch_lightning.metrics import Accuracy
 import torch
 import torch.nn.functional as F
+from pytorch_lightning.metrics import Accuracy
 
 from pl_bolts.models.self_supervised import SSLEvaluator
 

pl_bolts/models/self_supervised/swav/swav_finetuner.py

@@ -1,5 +1,5 @@
-from argparse import ArgumentParser
 import os
+from argparse import ArgumentParser
 
 import pytorch_lightning as pl
 

pl_bolts/models/self_supervised/swav/swav_module.py

@@ -1,17 +1,17 @@
 """
 Adapted from official swav implementation: https://github.com/facebookresearch/swav
 """
-from argparse import ArgumentParser
 import math
 import os
+from argparse import ArgumentParser
 from typing import Callable, Optional
 
 import numpy as np
 import pytorch_lightning as pl
-from pytorch_lightning.utilities import AMPType
 import torch
-from torch import nn
 import torch.distributed as dist
+from pytorch_lightning.utilities import AMPType
+from torch import nn
 from torch.optim.optimizer import Optimizer
 
 from pl_bolts.models.self_supervised.swav.swav_resnet import resnet18, resnet50

pl_bolts/models/vision/image_gpt/igpt_module.py

@@ -1,5 +1,5 @@
-from argparse import ArgumentParser
 import os
+from argparse import ArgumentParser
 
 import pytorch_lightning as pl
 import torch

pl_bolts/optimizers/lr_scheduler.py

@@ -1,6 +1,6 @@
 import math
-from typing import List
 import warnings
+from typing import List
 
 import torch.nn as nn
 from torch.optim import Adam, Optimizer

pl_bolts/utils/__init__.py

@@ -1,5 +1,5 @@
-from pytorch_lightning.utilities import _module_available
 import torch
+from pytorch_lightning.utilities import _module_available
 
 _NATIVE_AMP_AVAILABLE: bool = _module_available("torch.cuda.amp") and hasattr(torch.cuda.amp, "autocast")