.

Lightning-AI · Jan 1, 2021 · 9dd70b9 · 9dd70b9
1 parent e15d2de
commit 9dd70b9
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 4 deletions.
diff --git a/tests/base/model_train_steps.py b/tests/base/model_train_steps.py
@@ -17,8 +17,6 @@
 
 import torch
 
-from pytorch_lightning.core.step_result import TrainResult
-
 
 class TrainingStepVariations(ABC):
     """

diff --git a/tests/core/test_results.py b/tests/core/test_results.py
@@ -19,7 +19,7 @@
 import torch.distributed as dist
 import torch.multiprocessing as mp
 from pytorch_lightning import Trainer
-from pytorch_lightning.core.step_result import Result, TrainResult, EvalResult
+from pytorch_lightning.core.step_result import Result, EvalResult
 import tests.base.develop_utils as tutils
 
 from tests.base import EvalModelTemplate

diff --git a/tests/models/test_sync_batchnorm.py b/tests/models/test_sync_batchnorm.py
@@ -17,7 +17,6 @@
 import torch.nn.functional as F
 
 from pytorch_lightning import Trainer, seed_everything, LightningModule
-from pytorch_lightning.core.step_result import TrainResult
 from pytorch_lightning.utilities import FLOAT16_EPSILON
 from tests.base.datamodules import MNISTDataModule
 from tests.base.develop_utils import set_random_master_port
-Original file line number
+Diff line change
@@ Expand Up / @@ -17,8 +17,6 @@ @@
     import torch
-    from pytorch_lightning.core.step_result import TrainResult
     class TrainingStepVariations(ABC):
         """
@@ Expand Down @@