Check the evaluation events in the evaluation tests.
This commit is contained in:
parent
1230811c77
commit
92d622d27f
@ -1290,8 +1290,8 @@ class EstimatorEvaluateTest(test.TestCase):
|
||||
# Make sure nothing is stuck in limbo.
|
||||
writer_cache.FileWriterCache.clear()
|
||||
|
||||
# Get last Event written.
|
||||
if check_eventfile_for_keyword('image', est.model_dir):
|
||||
# Get last evaluation Event written.
|
||||
if check_eventfile_for_keyword('image', os.path.join(est.model_dir, 'eval')):
|
||||
return
|
||||
self.fail('{} should be part of reported summaries.'.format('image'))
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user