Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add test assertion (#9309)

This commit is contained in:
Carlos Mocholí 2021-09-06 18:06:26 +02:00 committed by GitHub
parent 98e2f56db0
commit 392c577825
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/trainer/loops/test_evaluation_loop.py
View File

@ -16,8 +16,9 @@ from unittest import mock
import torch
from torch.utils.data import DataLoader
from pytorch_lightning import LightningModule, Trainer
from pytorch_lightning import Trainer
from pytorch_lightning.loops import EvaluationEpochLoop
from pytorch_lightning.utilities.model_helpers import is_overridden
from tests.helpers.boring_model import BoringModel, RandomDataset
from tests.helpers.runif import RunIf
@ -120,8 +121,8 @@ def test_evaluation_loop_doesnt_store_outputs_if_epoch_end_not_overridden(tmpdir
super().on_advance_end()
model = TestModel()
# make sure this hook is not overridden
model.test_epoch_end = LightningModule.test_epoch_end
model.test_epoch_end = None
assert not is_overridden("test_epoch_end", model)
trainer = Trainer(default_root_dir=tmpdir, fast_dev_run=3)
trainer.test_loop.connect(TestLoop())