diff --git a/.circleci/config.yml b/.circleci/config.yml index 3041c184cc..35ccf246c0 100755 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -56,7 +56,7 @@ references: pip install -r requirements.txt --user sudo pip install -r docs/requirements.txt # sphinx-apidoc -o ./docs/source ./pytorch_lightning **/test_* --force --follow-links - cd docs; make clean ; make html + cd docs; make clean ; make html --debug --jobs 2 jobs: diff --git a/pytorch_lightning/trainer/evaluation_loop.py b/pytorch_lightning/trainer/evaluation_loop.py index a16dea78ab..a00a228ba8 100644 --- a/pytorch_lightning/trainer/evaluation_loop.py +++ b/pytorch_lightning/trainer/evaluation_loop.py @@ -132,7 +132,7 @@ import torch from torch.utils.data import DataLoader from tqdm.auto import tqdm -from pytorch_lightning import LightningModule +from pytorch_lightning.core.lightning import LightningModule from pytorch_lightning.utilities.debugging import MisconfigurationException try: diff --git a/pytorch_lightning/trainer/training_io.py b/pytorch_lightning/trainer/training_io.py index 68645fdfbe..9cfe053729 100644 --- a/pytorch_lightning/trainer/training_io.py +++ b/pytorch_lightning/trainer/training_io.py @@ -101,8 +101,8 @@ from typing import Union import torch import torch.distributed as dist +from pytorch_lightning.core.lightning import LightningModule from pytorch_lightning.loggers import LightningLoggerBase -from pytorch_lightning import LightningModule from pytorch_lightning.overrides.data_parallel import ( LightningDistributedDataParallel, LightningDataParallel, diff --git a/pytorch_lightning/trainer/training_loop.py b/pytorch_lightning/trainer/training_loop.py index 28761054b4..03df18510d 100644 --- a/pytorch_lightning/trainer/training_loop.py +++ b/pytorch_lightning/trainer/training_loop.py @@ -163,8 +163,8 @@ from typing import Union, List import numpy as np from torch.utils.data import DataLoader +from pytorch_lightning.core.lightning import LightningModule from pytorch_lightning.loggers import LightningLoggerBase -from pytorch_lightning import LightningModule from pytorch_lightning.utilities.debugging import MisconfigurationException from pytorch_lightning.callbacks.base import Callback