diff --git a/pytorch_lightning/profiler/__init__.py b/pytorch_lightning/profiler/__init__.py index fc61a2e3f9abe..e09a5ea11a084 100644 --- 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 index 0ea0e052d3da5..2bbcaa2e97cf3 100644 --- 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 index 769c3e2098794..8aeb687f1c927 100644 --- 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 index 95618ec5ca53d..b198d983cfe6d 100644 --- 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 index d9fd312ba9345..8cbae958fd483 100644 --- 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 index 49794c8b06b43..866bffcdd7441 100644 --- 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 index ef4ace7153e5a..98355a2d10c60 100644 --- 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 index 4e6f159054a2d..537dfd3e1006b 100644 --- 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 index fdaf234be1538..391c3ad4d2d04 100644 --- 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):