diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-10 19:50:28 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-10 19:53:39 -0700 |
| commit | e6830cdb06efe6f4cea2e4f30aa98f66ee1b305a (patch) | |
| tree | f51142af388f5e02ad636288c15921754c304322 /tensorflow/contrib/data | |
| parent | 497715e0a9bbb3c844a1902e319778cc30819f77 (diff) | |
Resolving a bug where regex pattern for errors was not matching in case the error message had multiple newline characters.
PiperOrigin-RevId: 212381070
Diffstat (limited to 'tensorflow/contrib/data')
| -rw-r--r-- | tensorflow/contrib/data/python/kernel_tests/test_utils.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/tensorflow/contrib/data/python/kernel_tests/test_utils.py b/tensorflow/contrib/data/python/kernel_tests/test_utils.py index 1def07179a..4c3353fe40 100644 --- a/tensorflow/contrib/data/python/kernel_tests/test_utils.py +++ b/tensorflow/contrib/data/python/kernel_tests/test_utils.py @@ -52,8 +52,11 @@ class DatasetTestBase(test.TestCase): dataset2, exception_class, replacements=None): - next1 = dataset1.make_one_shot_iterator().get_next() - next2 = dataset2.make_one_shot_iterator().get_next() + # We are defining next1 and next2 in the same line so that we get identical + # file:line_number in the error messages + # pylint: disable=line-too-long + next1, next2 = dataset1.make_one_shot_iterator().get_next(), dataset2.make_one_shot_iterator().get_next() + # pylint: enable=line-too-long with self.cached_session() as sess: try: sess.run(next1) |