diff options
| author |  Akshay Modi <nareshmodi@google.com> | 2018-04-03 10:04:47 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-03 10:09:05 -0700 |
| commit | 655b2663e7d2609b0f578b9ef3c1401de22dc5c2 (patch) | |
| tree | 61ac9426556a65f99edf99a23fa491d329d70016 /tensorflow/python/lib | |
| parent | cfc886ac6064a04c71dd6c52e8c21ebec91eae50 (diff) | |
Apply "Raise exception in SWIG on bad TF_Status" to base.i
Minor fixes to make this work.
PiperOrigin-RevId: 191457070
Diffstat (limited to 'tensorflow/python/lib')
| -rw-r--r-- | tensorflow/python/lib/io/tf_record.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/python/lib/io/tf_record.py b/tensorflow/python/lib/io/tf_record.py index 6fcf9c91d8..bf2d6f68b5 100644 --- a/tensorflow/python/lib/io/tf_record.py +++ b/tensorflow/python/lib/io/tf_record.py @@ -78,8 +78,7 @@ def tf_record_iterator(path, options=None): try: while True: try: - with errors.raise_exception_on_not_ok_status() as status: - reader.GetNext(status) + reader.GetNext() except errors.OutOfRangeError: break yield reader.record() |