fix docs (#982)

.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:
 

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:

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,

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