diff options
author | 2018-06-29 10:30:13 -0700 | |
---|---|---|
committer | 2018-06-29 10:32:46 -0700 | |
commit | 0979821324cbfce24e3162af9d5c0fea747e5ec0 (patch) | |
tree | d2ca434711d0178a4f1a52d6e555aafd6644ad9f /tensorflow/contrib/slim | |
parent | f139bc3f5cfb52882d9882d2acfdb818409209ce (diff) |
Add more helpful error messages when restoring from checkpoint fails.
PiperOrigin-RevId: 202668227
Diffstat (limited to 'tensorflow/contrib/slim')
-rw-r--r-- | tensorflow/contrib/slim/python/slim/evaluation_test.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/contrib/slim/python/slim/evaluation_test.py b/tensorflow/contrib/slim/python/slim/evaluation_test.py index 3d0308aaf3..2c97834523 100644 --- a/tensorflow/contrib/slim/python/slim/evaluation_test.py +++ b/tensorflow/contrib/slim/python/slim/evaluation_test.py @@ -33,7 +33,6 @@ from tensorflow.python.debug.lib import debug_data from tensorflow.python.debug.wrappers import hooks from tensorflow.python.framework import constant_op from tensorflow.python.framework import dtypes -from tensorflow.python.framework import errors from tensorflow.python.ops import control_flow_ops from tensorflow.python.ops import math_ops from tensorflow.python.ops import metrics @@ -242,7 +241,7 @@ class SingleEvaluationTest(test.TestCase): checkpoint_path = os.path.join(self.get_temp_dir(), 'this_file_doesnt_exist') log_dir = os.path.join(self.get_temp_dir(), 'error_raised') - with self.assertRaises(errors.NotFoundError): + with self.assertRaises(ValueError): evaluation.evaluate_once('', checkpoint_path, log_dir) def _prepareCheckpoint(self, checkpoint_path): |