Fix pre-commit isort failure on tests/trainer/*.py

arnaudgelas · arnaudgelas · commit f3d61ba4a433 · 2021-01-14T09:21:30.000+01:00
diff --git a/tests/trainer/data_flow/test_eval_loop_flow_1_0.py b/tests/trainer/data_flow/test_eval_loop_flow_1_0.py
@@ -17,10 +17,11 @@
 import os
 from unittest import mock
 
-import torch
 import pytest
-from pytorch_lightning.core.lightning import LightningModule
+import torch
+
 from pytorch_lightning import Trainer
+from pytorch_lightning.core.lightning import LightningModule
 from tests.base.deterministic_model import DeterministicModel
 
 
diff --git a/tests/trainer/data_flow/test_train_loop_flow_dict_1_0.py b/tests/trainer/data_flow/test_train_loop_flow_dict_1_0.py
@@ -17,10 +17,11 @@
 import os
 from unittest import mock
 
-from pytorch_lightning.core.lightning import LightningModule
+import torch
+
 from pytorch_lightning import Trainer
+from pytorch_lightning.core.lightning import LightningModule
 from tests.base.deterministic_model import DeterministicModel
-import torch
 
 
 @mock.patch.dict(os.environ, {"PL_DEV_DEBUG": "1"})
diff --git a/tests/trainer/data_flow/test_train_loop_flow_scalar_1_0.py b/tests/trainer/data_flow/test_train_loop_flow_scalar_1_0.py
@@ -17,12 +17,13 @@
 import os
 from unittest import mock
 
-from pytorch_lightning.core.lightning import LightningModule
 import pytest
+import torch
+
 from pytorch_lightning import Trainer
-from tests.base.deterministic_model import DeterministicModel
+from pytorch_lightning.core.lightning import LightningModule
 from tests.base.boring_model import BoringModel
-import torch
+from tests.base.deterministic_model import DeterministicModel
 
 
 @mock.patch.dict(os.environ, {"PL_DEV_DEBUG": "1"})
diff --git a/tests/trainer/dynamic_args/test_multiple_eval_dataloaders.py b/tests/trainer/dynamic_args/test_multiple_eval_dataloaders.py
@@ -11,11 +11,12 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from pytorch_lightning import Trainer
-from tests.base.boring_model import BoringModel
 import torch
 from torch.utils.data import Dataset
 
+from pytorch_lightning import Trainer
+from tests.base.boring_model import BoringModel
+
 
 class RandomDatasetA(Dataset):
     def __init__(self, size, length):
diff --git a/tests/trainer/dynamic_args/test_multiple_optimizers.py b/tests/trainer/dynamic_args/test_multiple_optimizers.py
@@ -11,9 +11,10 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+import torch
+
 from pytorch_lightning import Trainer
 from tests.base.boring_model import BoringModel
-import torch
 
 
 def test_multiple_optimizers(tmpdir):
diff --git a/tests/trainer/flags/test_overfit_batches.py b/tests/trainer/flags/test_overfit_batches.py
@@ -11,10 +11,11 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-import torch
 import pytest
-from tests.base.boring_model import BoringModel, RandomDataset
+import torch
+
 from pytorch_lightning import Trainer
+from tests.base.boring_model import BoringModel, RandomDataset
 
 
 def test_overfit_multiple_val_loaders(tmpdir):
diff --git a/tests/trainer/flags/test_val_check_interval.py b/tests/trainer/flags/test_val_check_interval.py
@@ -12,8 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import pytest
-from tests.base import SimpleModule
+
 from pytorch_lightning.trainer import Trainer
+from tests.base import SimpleModule
 
 
 @pytest.mark.parametrize('max_epochs', [1, 2, 3])
diff --git a/tests/trainer/legacy_deprecate_flow_log_tests/test_eval_loop_dict_return.py b/tests/trainer/legacy_deprecate_flow_log_tests/test_eval_loop_dict_return.py
@@ -14,8 +14,8 @@
 """
 Tests to ensure that the training loop works with a dict
 """
-from pytorch_lightning.core.lightning import LightningModule
 from pytorch_lightning import Trainer
+from pytorch_lightning.core.lightning import LightningModule
 from tests.base.deterministic_model import DeterministicModel
 
 
diff --git a/tests/trainer/legacy_deprecate_flow_log_tests/test_trainer_steps_scalar_return.py b/tests/trainer/legacy_deprecate_flow_log_tests/test_trainer_steps_scalar_return.py
@@ -17,12 +17,12 @@
 import os
 from unittest import mock
 
-import torch
 import pytest
+import torch
 
 from pytorch_lightning import Trainer
-from tests.base.deterministic_model import DeterministicModel
 from tests.base import BoringModel
+from tests.base.deterministic_model import DeterministicModel
 
 
 def test_training_step_scalar(tmpdir):
diff --git a/tests/trainer/logging_tests/test_distributed_logging.py b/tests/trainer/logging_tests/test_distributed_logging.py
@@ -12,12 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import platform
+from unittest import mock
+
 import pytest
 import torch
-from tests.base import BoringModel
-import platform
+
 from pytorch_lightning import Trainer
-from unittest import mock
+from tests.base import BoringModel
 
 
 class TestModel(BoringModel):
diff --git a/tests/trainer/model_hooks/test_model_hooks.py b/tests/trainer/model_hooks/test_model_hooks.py
@@ -11,10 +11,11 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from tests.base.boring_model import BoringModel
-from pytorch_lightning import Trainer
 from unittest import mock
 
+from pytorch_lightning import Trainer
+from tests.base.boring_model import BoringModel
+
 
 @mock.patch('pytorch_lightning.core.hooks.ModelHooks.on_validation_model_eval')
 @mock.patch('pytorch_lightning.core.hooks.ModelHooks.on_validation_model_train')
diff --git a/tests/trainer/optimization/test_manual_optimization.py b/tests/trainer/optimization/test_manual_optimization.py
@@ -21,7 +21,7 @@
 import torch.distributed as torch_distrib
 import torch.nn.functional as F
 
-from pytorch_lightning import Trainer, seed_everything
+from pytorch_lightning import seed_everything, Trainer
 from pytorch_lightning.utilities import _APEX_AVAILABLE
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 from tests.base.boring_model import BoringModel
diff --git a/tests/trainer/properties/log_dir.py b/tests/trainer/properties/log_dir.py
@@ -12,8 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import os
-from tests.base.boring_model import BoringModel
+
 from pytorch_lightning import Trainer
+from tests.base.boring_model import BoringModel
 
 
 def test_logdir(tmpdir):
diff --git a/tests/trainer/test_config_validator.py b/tests/trainer/test_config_validator.py
@@ -18,7 +18,6 @@
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 from tests.base import EvalModelTemplate
 
-
 # TODO: add matching messages
 
 
diff --git a/tests/trainer/test_lr_finder.py b/tests/trainer/test_lr_finder.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 import os
 from copy import deepcopy
+
 import pytest
 import torch
 
diff --git a/tests/trainer/test_states.py b/tests/trainer/test_states.py
@@ -13,8 +13,8 @@
 # limitations under the License.
 import pytest
 
-from pytorch_lightning import Trainer, Callback
-from pytorch_lightning.trainer.states import TrainerState, trainer_state
+from pytorch_lightning import Callback, Trainer
+from pytorch_lightning.trainer.states import trainer_state, TrainerState
 from tests.base import EvalModelTemplate
 
 
diff --git a/tests/trainer/test_supporters.py b/tests/trainer/test_supporters.py
@@ -15,10 +15,15 @@
 
 import pytest
 import torch
-
 from torch.utils.data import TensorDataset
+
 from pytorch_lightning.trainer.supporters import (
-    CycleIterator, CombinedLoader, CombinedDataset, CombinedLoaderIterator, TensorRunningAccum)
+    CombinedDataset,
+    CombinedLoader,
+    CombinedLoaderIterator,
+    CycleIterator,
+    TensorRunningAccum,
+)
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 
 
diff --git a/tests/trainer/test_trainer_tricks.py b/tests/trainer/test_trainer_tricks.py
@@ -13,13 +13,14 @@
 # limitations under the License.
 import os
 from copy import deepcopy
+
 import pytest
 import torch
-from torch.utils.data import RandomSampler, SequentialSampler, DataLoader
+from torch.utils.data import DataLoader, RandomSampler, SequentialSampler
 
 import tests.base.develop_utils as tutils
 from pytorch_lightning import Trainer
-from pytorch_lightning.utilities import AMPType, _NATIVE_AMP_AVAILABLE
+from pytorch_lightning.utilities import _NATIVE_AMP_AVAILABLE, AMPType
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 from tests.base import EvalModelTemplate
 from tests.base.datamodules import MNISTDataModule
diff --git a/tests/trainer/warnings_tests/test_flow_warnings.py b/tests/trainer/warnings_tests/test_flow_warnings.py
@@ -12,11 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 import os
+import warnings
 from unittest import mock
 
-from tests.base.boring_model import BoringModel
 from pytorch_lightning import Trainer
-import warnings
+from tests.base.boring_model import BoringModel
 
 
 class TestModel(BoringModel):
