isort

Lightning-AI · Mar 2, 2021 · a8cd1d0 · a8cd1d0
1 parent a3307fa
commit a8cd1d0
Show file tree

Hide file tree

Showing 9 changed files with 9 additions and 14 deletions.
diff --git a/pytorch_lightning/profiler/__init__.py b/pytorch_lightning/profiler/__init__.py
@@ -194,12 +194,7 @@ def custom_processing_step(self, data):
 
 """
 
-from pytorch_lightning.profiler.profilers import (
-    AdvancedProfiler,
-    BaseProfiler,
-    PassThroughProfiler,
-    SimpleProfiler,
-)
+from pytorch_lightning.profiler.profilers import AdvancedProfiler, BaseProfiler, PassThroughProfiler, SimpleProfiler
 from pytorch_lightning.profiler.pytorch import PyTorchProfiler
 
 __all__ = [

diff --git a/tests/accelerators/test_ddp_spawn.py b/tests/accelerators/test_ddp_spawn.py
@@ -20,8 +20,8 @@
 from pytorch_lightning.trainer.states import TrainerState
 from tests.helpers import BoringModel
 from tests.helpers.datamodules import ClassifDataModule
-from tests.helpers.simple_models import ClassificationModel
 from tests.helpers.runif import RunIf
+from tests.helpers.simple_models import ClassificationModel
 
 
 @RunIf(min_gpus=2)

diff --git a/tests/accelerators/test_dp.py b/tests/accelerators/test_dp.py
@@ -21,8 +21,8 @@
 from pytorch_lightning.core import memory
 from tests.helpers import BoringModel
 from tests.helpers.datamodules import ClassifDataModule
-from tests.helpers.simple_models import ClassificationModel
 from tests.helpers.runif import RunIf
+from tests.helpers.simple_models import ClassificationModel
 
 PRETEND_N_OF_GPUS = 16
 

diff --git a/tests/callbacks/test_early_stopping.py b/tests/callbacks/test_early_stopping.py
@@ -26,8 +26,8 @@
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 from tests.helpers import BoringModel
 from tests.helpers.datamodules import ClassifDataModule
-from tests.helpers.simple_models import ClassificationModel
 from tests.helpers.runif import RunIf
+from tests.helpers.simple_models import ClassificationModel
 
 
 class EarlyStoppingTestRestore(EarlyStopping):

diff --git a/tests/callbacks/test_quantization.py b/tests/callbacks/test_quantization.py
@@ -21,8 +21,8 @@
 from pytorch_lightning.metrics.functional.mean_relative_error import mean_relative_error
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 from tests.helpers.datamodules import RegressDataModule
-from tests.helpers.simple_models import RegressionModel
 from tests.helpers.runif import RunIf
+from tests.helpers.simple_models import RegressionModel
 
 
 @pytest.mark.parametrize("observe", ['average', pytest.param('histogram', marks=RunIf(min_torch="1.5"))])

diff --git a/tests/core/test_datamodules.py b/tests/core/test_datamodules.py
@@ -26,8 +26,8 @@
 from pytorch_lightning.utilities.model_helpers import is_overridden
 from tests.helpers import BoringDataModule, BoringModel
 from tests.helpers.datamodules import ClassifDataModule
-from tests.helpers.simple_models import ClassificationModel
 from tests.helpers.runif import RunIf
+from tests.helpers.simple_models import ClassificationModel
 from tests.helpers.utils import reset_seed, set_random_master_port
 
 

diff --git a/tests/models/test_cpu.py b/tests/models/test_cpu.py
@@ -22,8 +22,8 @@
 from pytorch_lightning.trainer.states import TrainerState
 from tests.helpers import BoringModel
 from tests.helpers.datamodules import ClassifDataModule
-from tests.helpers.simple_models import ClassificationModel
 from tests.helpers.runif import RunIf
+from tests.helpers.simple_models import ClassificationModel
 
 
 def test_cpu_slurm_save_load(tmpdir):

diff --git a/tests/models/test_gpu.py b/tests/models/test_gpu.py
@@ -25,8 +25,8 @@
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 from tests.helpers import BoringModel
 from tests.helpers.datamodules import ClassifDataModule
-from tests.helpers.simple_models import ClassificationModel
 from tests.helpers.runif import RunIf
+from tests.helpers.simple_models import ClassificationModel
 
 PRETEND_N_OF_GPUS = 16
 

diff --git a/tests/models/test_restore.py b/tests/models/test_restore.py
@@ -30,8 +30,8 @@
 from pytorch_lightning.trainer.states import RunningStage, TrainerState
 from tests.helpers import BoringModel
 from tests.helpers.datamodules import ClassifDataModule
-from tests.helpers.simple_models import ClassificationModel
 from tests.helpers.runif import RunIf
+from tests.helpers.simple_models import ClassificationModel
 
 
 class ModelTrainerPropertyParity(Callback):