From 0b7c552df4edb86bc67447659fdf8486151733d2 Mon Sep 17 00:00:00 2001 From: Akihiro Nitta Date: Sun, 3 Jan 2021 19:41:05 +0900 Subject: [PATCH] Apply isort --- pl_bolts/callbacks/data_monitor.py | 4 ++-- pl_bolts/callbacks/ssl_online.py | 2 +- pl_bolts/callbacks/variational.py | 2 +- pl_bolts/callbacks/vision/confused_logit.py | 2 +- pl_bolts/callbacks/vision/image_generation.py | 2 +- pl_bolts/datamodules/async_dataloader.py | 2 +- pl_bolts/datamodules/kitti_datamodule.py | 2 +- pl_bolts/datamodules/sklearn_datamodule.py | 2 +- pl_bolts/datamodules/stl10_datamodule.py | 2 +- pl_bolts/datamodules/vision_datamodule.py | 4 ++-- pl_bolts/datamodules/vocdetection_datamodule.py | 4 ++-- pl_bolts/datasets/base_dataset.py | 4 ++-- pl_bolts/datasets/imagenet_dataset.py | 2 +- pl_bolts/models/autoencoders/components.py | 2 +- pl_bolts/models/gans/basic/components.py | 2 +- pl_bolts/models/mnist_module.py | 2 +- pl_bolts/models/regression/logistic_regression.py | 2 +- pl_bolts/models/rl/common/agents.py | 2 +- pl_bolts/models/rl/common/gym_wrappers.py | 2 +- pl_bolts/models/rl/dqn_model.py | 4 ++-- pl_bolts/models/rl/reinforce_model.py | 4 ++-- pl_bolts/models/rl/vanilla_policy_gradient_model.py | 4 ++-- pl_bolts/models/self_supervised/amdim/amdim_module.py | 2 +- pl_bolts/models/self_supervised/amdim/networks.py | 2 +- pl_bolts/models/self_supervised/byol/byol_module.py | 2 +- pl_bolts/models/self_supervised/cpc/cpc_finetuner.py | 2 +- pl_bolts/models/self_supervised/cpc/cpc_module.py | 4 ++-- pl_bolts/models/self_supervised/cpc/networks.py | 2 +- pl_bolts/models/self_supervised/moco/moco2_module.py | 2 +- pl_bolts/models/self_supervised/simclr/simclr_finetuner.py | 2 +- pl_bolts/models/self_supervised/simclr/simclr_module.py | 6 +++--- pl_bolts/models/self_supervised/ssl_finetuner.py | 2 +- pl_bolts/models/self_supervised/swav/swav_finetuner.py | 2 +- pl_bolts/models/self_supervised/swav/swav_module.py | 6 +++--- pl_bolts/models/vision/image_gpt/igpt_module.py | 2 +- pl_bolts/optimizers/lr_scheduler.py | 2 +- pl_bolts/utils/__init__.py | 2 +- pl_bolts/utils/arguments.py | 2 +- pl_bolts/utils/warnings.py | 2 +- tests/callbacks/test_data_monitor.py | 4 ++-- tests/datamodules/test_datamodules.py | 4 ++-- tests/models/self_supervised/test_models.py | 2 +- tests/models/test_autoencoders.py | 2 +- tests/optimizers/test_lr_scheduler.py | 2 +- 44 files changed, 58 insertions(+), 58 deletions(-) diff --git a/pl_bolts/callbacks/data_monitor.py b/pl_bolts/callbacks/data_monitor.py index 81f987c534..f98f3662b6 100644 --- a/pl_bolts/callbacks/data_monitor.py +++ b/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: diff --git a/pl_bolts/callbacks/ssl_online.py b/pl_bolts/callbacks/ssl_online.py index fd3c842456..ee27196ae8 100644 --- a/pl_bolts/callbacks/ssl_online.py +++ b/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 diff --git a/pl_bolts/callbacks/variational.py b/pl_bolts/callbacks/variational.py index 5f40a9fc98..61aea5d86e 100644 --- a/pl_bolts/callbacks/variational.py +++ b/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 diff --git a/pl_bolts/callbacks/vision/confused_logit.py b/pl_bolts/callbacks/vision/confused_logit.py index 2ad75be9ad..235dc95f1f 100644 --- a/pl_bolts/callbacks/vision/confused_logit.py +++ b/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 diff --git a/pl_bolts/callbacks/vision/image_generation.py b/pl_bolts/callbacks/vision/image_generation.py index cde6d39ee4..ac71c0dfd0 100644 --- a/pl_bolts/callbacks/vision/image_generation.py +++ b/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 diff --git a/pl_bolts/datamodules/async_dataloader.py b/pl_bolts/datamodules/async_dataloader.py index 58c0b12f59..7ded9d9ef1 100644 --- a/pl_bolts/datamodules/async_dataloader.py +++ b/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 diff --git a/pl_bolts/datamodules/kitti_datamodule.py b/pl_bolts/datamodules/kitti_datamodule.py index 56bd5ec6a9..433e7fffed 100644 --- a/pl_bolts/datamodules/kitti_datamodule.py +++ b/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 diff --git a/pl_bolts/datamodules/sklearn_datamodule.py b/pl_bolts/datamodules/sklearn_datamodule.py index 2ba0aa9976..ed262b10c8 100644 --- a/pl_bolts/datamodules/sklearn_datamodule.py +++ b/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 diff --git a/pl_bolts/datamodules/stl10_datamodule.py b/pl_bolts/datamodules/stl10_datamodule.py index 80210ce538..c666db9b9b 100644 --- a/pl_bolts/datamodules/stl10_datamodule.py +++ b/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 diff --git a/pl_bolts/datamodules/vision_datamodule.py b/pl_bolts/datamodules/vision_datamodule.py index 76a53c9cb9..2144f0f509 100644 --- a/pl_bolts/datamodules/vision_datamodule.py +++ b/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 diff --git a/pl_bolts/datamodules/vocdetection_datamodule.py b/pl_bolts/datamodules/vocdetection_datamodule.py index 2ee27f63f7..a2087f9448 100644 --- a/pl_bolts/datamodules/vocdetection_datamodule.py +++ b/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 diff --git a/pl_bolts/datasets/base_dataset.py b/pl_bolts/datasets/base_dataset.py index 198efb6d08..0c24395063 100644 --- a/pl_bolts/datasets/base_dataset.py +++ b/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 diff --git a/pl_bolts/datasets/imagenet_dataset.py b/pl_bolts/datasets/imagenet_dataset.py index 677a6695bb..91ff3742e6 100644 --- a/pl_bolts/datasets/imagenet_dataset.py +++ b/pl_bolts/datasets/imagenet_dataset.py @@ -1,4 +1,3 @@ -from contextlib import contextmanager import gzip import hashlib import os @@ -6,6 +5,7 @@ import tarfile import tempfile import zipfile +from contextlib import contextmanager import numpy as np import torch diff --git a/pl_bolts/models/autoencoders/components.py b/pl_bolts/models/autoencoders/components.py index b3193d968b..c8b1a45548 100644 --- a/pl_bolts/models/autoencoders/components.py +++ b/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): diff --git a/pl_bolts/models/gans/basic/components.py b/pl_bolts/models/gans/basic/components.py index 0cf5caadae..25812fab4e 100644 --- a/pl_bolts/models/gans/basic/components.py +++ b/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): diff --git a/pl_bolts/models/mnist_module.py b/pl_bolts/models/mnist_module.py index f6437d30a6..74a467b48a 100644 --- a/pl_bolts/models/mnist_module.py +++ b/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 diff --git a/pl_bolts/models/regression/logistic_regression.py b/pl_bolts/models/regression/logistic_regression.py index e6769fd3f0..ea9f1dcc24 100644 --- a/pl_bolts/models/regression/logistic_regression.py +++ b/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 diff --git a/pl_bolts/models/rl/common/agents.py b/pl_bolts/models/rl/common/agents.py index dd07805d8d..d9f4d9d063 100644 --- a/pl_bolts/models/rl/common/agents.py +++ b/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): diff --git a/pl_bolts/models/rl/common/gym_wrappers.py b/pl_bolts/models/rl/common/gym_wrappers.py index 911e4bb3e0..0317a5820d 100644 --- a/pl_bolts/models/rl/common/gym_wrappers.py +++ b/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 diff --git a/pl_bolts/models/rl/dqn_model.py b/pl_bolts/models/rl/dqn_model.py index 38bf99829e..6f5e958a36 100644 --- a/pl_bolts/models/rl/dqn_model.py +++ b/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 diff --git a/pl_bolts/models/rl/reinforce_model.py b/pl_bolts/models/rl/reinforce_model.py index ac109e6550..645d7b18b3 100644 --- a/pl_bolts/models/rl/reinforce_model.py +++ b/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 diff --git a/pl_bolts/models/rl/vanilla_policy_gradient_model.py b/pl_bolts/models/rl/vanilla_policy_gradient_model.py index 3f0d2f9da0..0dcfb5a37b 100644 --- a/pl_bolts/models/rl/vanilla_policy_gradient_model.py +++ b/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 diff --git a/pl_bolts/models/self_supervised/amdim/amdim_module.py b/pl_bolts/models/self_supervised/amdim/amdim_module.py index 3ffd2d1499..31192cdaae 100644 --- a/pl_bolts/models/self_supervised/amdim/amdim_module.py +++ b/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 diff --git a/pl_bolts/models/self_supervised/amdim/networks.py b/pl_bolts/models/self_supervised/amdim/networks.py index f73765852d..fe330300a6 100644 --- a/pl_bolts/models/self_supervised/amdim/networks.py +++ b/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): diff --git a/pl_bolts/models/self_supervised/byol/byol_module.py b/pl_bolts/models/self_supervised/byol/byol_module.py index 75a01c50e8..d212efb0a5 100644 --- a/pl_bolts/models/self_supervised/byol/byol_module.py +++ b/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 diff --git a/pl_bolts/models/self_supervised/cpc/cpc_finetuner.py b/pl_bolts/models/self_supervised/cpc/cpc_finetuner.py index 7b4bf3aead..a1decdf7c6 100644 --- a/pl_bolts/models/self_supervised/cpc/cpc_finetuner.py +++ b/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 diff --git a/pl_bolts/models/self_supervised/cpc/cpc_module.py b/pl_bolts/models/self_supervised/cpc/cpc_module.py index 5331c08cc6..23f8f50c34 100644 --- a/pl_bolts/models/self_supervised/cpc/cpc_module.py +++ b/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 diff --git a/pl_bolts/models/self_supervised/cpc/networks.py b/pl_bolts/models/self_supervised/cpc/networks.py index 79a515718b..bea33e4684 100644 --- a/pl_bolts/models/self_supervised/cpc/networks.py +++ b/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): diff --git a/pl_bolts/models/self_supervised/moco/moco2_module.py b/pl_bolts/models/self_supervised/moco/moco2_module.py index 957f64a6bd..56545ebe6c 100644 --- a/pl_bolts/models/self_supervised/moco/moco2_module.py +++ b/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 diff --git a/pl_bolts/models/self_supervised/simclr/simclr_finetuner.py b/pl_bolts/models/self_supervised/simclr/simclr_finetuner.py index b70cd55ec3..5e720e8f82 100644 --- a/pl_bolts/models/self_supervised/simclr/simclr_finetuner.py +++ b/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 diff --git a/pl_bolts/models/self_supervised/simclr/simclr_module.py b/pl_bolts/models/self_supervised/simclr/simclr_module.py index bc35042598..8e68eabf2b 100644 --- a/pl_bolts/models/self_supervised/simclr/simclr_module.py +++ b/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 diff --git a/pl_bolts/models/self_supervised/ssl_finetuner.py b/pl_bolts/models/self_supervised/ssl_finetuner.py index f4b3bdbb6c..31a3570542 100644 --- a/pl_bolts/models/self_supervised/ssl_finetuner.py +++ b/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 diff --git a/pl_bolts/models/self_supervised/swav/swav_finetuner.py b/pl_bolts/models/self_supervised/swav/swav_finetuner.py index c98153896a..7555121d9b 100644 --- a/pl_bolts/models/self_supervised/swav/swav_finetuner.py +++ b/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 diff --git a/pl_bolts/models/self_supervised/swav/swav_module.py b/pl_bolts/models/self_supervised/swav/swav_module.py index ad71ec4cf4..e081aa9ff9 100644 --- a/pl_bolts/models/self_supervised/swav/swav_module.py +++ b/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 diff --git a/pl_bolts/models/vision/image_gpt/igpt_module.py b/pl_bolts/models/vision/image_gpt/igpt_module.py index eea452b788..ad2d70c331 100644 --- a/pl_bolts/models/vision/image_gpt/igpt_module.py +++ b/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 diff --git a/pl_bolts/optimizers/lr_scheduler.py b/pl_bolts/optimizers/lr_scheduler.py index f93ad213aa..7a7f4ba2b1 100644 --- a/pl_bolts/optimizers/lr_scheduler.py +++ b/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 diff --git a/pl_bolts/utils/__init__.py b/pl_bolts/utils/__init__.py index b0b3cf1ce8..88aaa1b583 100644 --- a/pl_bolts/utils/__init__.py +++ b/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") diff --git a/pl_bolts/utils/arguments.py b/pl_bolts/utils/arguments.py index 4f5299101d..76d5659e7c 100644 --- a/pl_bolts/utils/arguments.py +++ b/pl_bolts/utils/arguments.py @@ -1,6 +1,6 @@ +import inspect from argparse import ArgumentParser, Namespace from dataclasses import dataclass -import inspect from typing import Any, Dict, List, Optional import pytorch_lightning as pl diff --git a/pl_bolts/utils/warnings.py b/pl_bolts/utils/warnings.py index b041d1f68d..d3f8c6a4e2 100644 --- a/pl_bolts/utils/warnings.py +++ b/pl_bolts/utils/warnings.py @@ -1,6 +1,6 @@ import os -from typing import Callable, Optional import warnings +from typing import Callable, Optional MISSING_PACKAGE_WARNINGS = {} diff --git a/tests/callbacks/test_data_monitor.py b/tests/callbacks/test_data_monitor.py index 6173a73083..a7cc4b8728 100644 --- a/tests/callbacks/test_data_monitor.py +++ b/tests/callbacks/test_data_monitor.py @@ -2,10 +2,10 @@ from unittest.mock import call import pytest -from pytorch_lightning import Trainer -from pytorch_lightning.loggers import LoggerCollection, TensorBoardLogger import torch import torch.nn as nn +from pytorch_lightning import Trainer +from pytorch_lightning.loggers import LoggerCollection, TensorBoardLogger from pl_bolts.callbacks import ModuleDataMonitor, TrainingDataMonitor from pl_bolts.models import LitMNIST diff --git a/tests/datamodules/test_datamodules.py b/tests/datamodules/test_datamodules.py index 95a9d544b0..2fc6a4c3df 100644 --- a/tests/datamodules/test_datamodules.py +++ b/tests/datamodules/test_datamodules.py @@ -1,9 +1,9 @@ -from pathlib import Path import uuid +from pathlib import Path -from PIL import Image import pytest import torch +from PIL import Image from pl_bolts.datamodules import ( BinaryMNISTDataModule, diff --git a/tests/models/self_supervised/test_models.py b/tests/models/self_supervised/test_models.py index 375d4c4afb..6ef6d3f1b1 100644 --- a/tests/models/self_supervised/test_models.py +++ b/tests/models/self_supervised/test_models.py @@ -1,7 +1,7 @@ import pytest import pytorch_lightning as pl -from pytorch_lightning import seed_everything import torch +from pytorch_lightning import seed_everything from pl_bolts.datamodules import CIFAR10DataModule from pl_bolts.models.self_supervised import AMDIM, BYOL, CPCV2, MocoV2, SimCLR, SwAV diff --git a/tests/models/test_autoencoders.py b/tests/models/test_autoencoders.py index cb68ce0c91..d08a35e92e 100644 --- a/tests/models/test_autoencoders.py +++ b/tests/models/test_autoencoders.py @@ -1,7 +1,7 @@ import pytest import pytorch_lightning as pl -from pytorch_lightning import seed_everything import torch +from pytorch_lightning import seed_everything from pl_bolts.datamodules import CIFAR10DataModule from pl_bolts.models.autoencoders import AE, resnet18_decoder, resnet18_encoder, resnet50_encoder, VAE diff --git a/tests/optimizers/test_lr_scheduler.py b/tests/optimizers/test_lr_scheduler.py index 7fc533a071..59646a8238 100644 --- a/tests/optimizers/test_lr_scheduler.py +++ b/tests/optimizers/test_lr_scheduler.py @@ -1,9 +1,9 @@ import math import numpy as np -from pytorch_lightning import seed_everything import torch import torch.nn.functional as F +from pytorch_lightning import seed_everything from torch.optim import SGD from torch.optim.lr_scheduler import _LRScheduler