wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-12 12:40:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix tests (#938)

Browse Source 
* fix tests

* fix tests
This commit is contained in:
William Falcon
2020-02-25 08:53:33 -05:00
committed by GitHub
parent 6b667b1237
commit ceec51d96c
2 changed files with 2 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/trainer/evaluation_loop.py
+2 -2
View File
@@ -301,8 +301,8 @@ class TrainerEvaluationLoopMixin(ABC):
def run_evaluation(self, test=False):
# when testing make sure user defined a test step
if test and not (self.is_overriden('test_step') or self.is_overriden('test_end')):
m = '''You called `.test()` without defining model's `.test_step()` or `.test_end()`.
if test and not self.is_overriden('test_step'):
m = '''You called `.test()` without defining model's `.test_step()`.
Please define and try again'''
raise MisconfigurationException(m)