diff options
| author |  Fei Hu <hufei68@gmail.com> | 2018-09-22 01:16:14 -0700 |
|---|---|---|
| committer | Fei Hu <hufei68@gmail.com> | 2018-09-22 01:16:14 -0700 |
| commit | 6ce84a89001df281f6100affa85fa7b8eb56c67f (patch) | |
| tree | 36ffd91683cb0ef0f5303784871a2b74c25d7cbd /tensorflow/python/framework | |
| parent | a5752eb9cb266262f3b7a289f12c21e268b3041d (diff) | |
Disable the logging for OutofRangeError in test
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 d63abd7f01..24dea8c336 100644 --- a/tensorflow/python/framework/test_util.py +++ b/tensorflow/python/framework/test_util.py @@ -877,7 +877,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 |