From e16781fad4e4a771d23210aa866d78c8518910b2 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 12 May 2022 09:42:51 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- pl_examples/loop_examples/kfold.py | 2 +- pytorch_lightning/loops/__init__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pl_examples/loop_examples/kfold.py b/pl_examples/loop_examples/kfold.py index a6186d30ea4f90..190115c8b6b0ab 100644 --- a/pl_examples/loop_examples/kfold.py +++ b/pl_examples/loop_examples/kfold.py @@ -31,8 +31,8 @@ from pl_examples.basic_examples.mnist_examples.image_classifier_4_lightning_module import ImageClassifier from pytorch_lightning import LightningDataModule, seed_everything, Trainer from pytorch_lightning.core.lightning import LightningModule -from pytorch_lightning.loops.loop import Loop from pytorch_lightning.loops.fit_loop import FitLoop +from pytorch_lightning.loops.loop import Loop from pytorch_lightning.trainer.states import TrainerFn ############################################################################################# diff --git a/pytorch_lightning/loops/__init__.py b/pytorch_lightning/loops/__init__.py index dcfcbc81950b52..1f473da07169c7 100644 --- a/pytorch_lightning/loops/__init__.py +++ b/pytorch_lightning/loops/__init__.py @@ -12,10 +12,10 @@ # See the License for the specific language governing permissions and # limitations under the License. -from pytorch_lightning.loops.loop import Loop # noqa: F401 from pytorch_lightning.loops.batch import ManualOptimization # noqa: F401 from pytorch_lightning.loops.batch import TrainingBatchLoop # noqa: F401 from pytorch_lightning.loops.dataloader import DataLoaderLoop, EvaluationLoop, PredictionLoop # noqa: F401 from pytorch_lightning.loops.epoch import EvaluationEpochLoop, PredictionEpochLoop, TrainingEpochLoop # noqa: F401 from pytorch_lightning.loops.fit_loop import FitLoop # noqa: F401 +from pytorch_lightning.loops.loop import Loop # noqa: F401 from pytorch_lightning.loops.optimization.optimizer_loop import OptimizerLoop # noqa: F401