diff options
author | 2018-09-24 08:36:23 -0700 | |
---|---|---|
committer | 2018-09-24 08:36:23 -0700 | |
commit | 7f71957f3818eff38671b758915a0a5c819c27e6 (patch) | |
tree | e7be33b81210f77a12f8c499ba7267e4467b7688 /tensorflow/python/framework | |
parent | 2aa490ea0327b7d01d966407eab88a2255669526 (diff) | |
parent | 6ce84a89001df281f6100affa85fa7b8eb56c67f (diff) |
Merge pull request #22461 from feihugis:Issue-12414_Error_Logging_Session
PiperOrigin-RevId: 214271324
Diffstat (limited to 'tensorflow/python/framework')
-rw-r--r-- | tensorflow/python/framework/test_util.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/python/framework/test_util.py b/tensorflow/python/framework/test_util.py index 68b7b323d5..e55a1b84ee 100644 --- a/tensorflow/python/framework/test_util.py +++ b/tensorflow/python/framework/test_util.py @@ -890,7 +890,11 @@ class ErrorLoggingSession(session.Session): try: return super(ErrorLoggingSession, self).run(*args, **kwargs) except Exception as e: # pylint: disable=broad-except - logging.error(str(e)) + # Note: disable the logging for OutOfRangeError, which makes the output + # of tf.data tests hard to read, because OutOfRangeError is used as the + # signal completion + if not isinstance(e, errors.OutOfRangeError): + logging.error(str(e)) raise |