aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/framework
diff options
context:
space:
mode:
authorGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-24 08:36:23 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-24 08:36:23 -0700
commit7f71957f3818eff38671b758915a0a5c819c27e6 (patch)
treee7be33b81210f77a12f8c499ba7267e4467b7688 /tensorflow/python/framework
parent2aa490ea0327b7d01d966407eab88a2255669526 (diff)
parent6ce84a89001df281f6100affa85fa7b8eb56c67f (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.py6
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