diff options
Diffstat (limited to 'tensorflow/python/lib/io/tf_record.py')
| -rw-r--r-- | tensorflow/python/lib/io/tf_record.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/python/lib/io/tf_record.py b/tensorflow/python/lib/io/tf_record.py index 3d0cdc2153..df19010068 100644 --- a/tensorflow/python/lib/io/tf_record.py +++ b/tensorflow/python/lib/io/tf_record.py @@ -121,6 +121,11 @@ class TFRecordWriter(object): """ self._writer.WriteRecord(record) + def flush(self): + """Flush the file.""" + with errors.raise_exception_on_not_ok_status() as status: + self._writer.Flush(status) + def close(self): """Close the file.""" with errors.raise_exception_on_not_ok_status() as status: |