fix docs (#982)

Lightning-AI · Feb 28, 2020 · 479a35d · 479a35d
1 parent 1ae16bd
commit 479a35d
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -132,7 +132,7 @@
 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
@@ -101,8 +101,8 @@
 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
@@ -163,8 +163,8 @@ def training_step(self, batch, batch_idx):
 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