Skip to content

GitLab

Commit c455baa8 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 475325492
parent c2697824
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/core/file_writers.py
View file @ c455baa8
...@@ -17,7 +17,6 @@ ...@@ -17,7 +17,6 @@
import io import io
from typing import Optional, Sequence, Union from typing import Optional, Sequence, Union
import riegeli
import tensorflow as tf import tensorflow as tf
...@@ -75,5 +74,6 @@ def _write_riegeli(examples: Sequence[Union[tf.train.Example, ...@@ -75,5 +74,6 @@ def _write_riegeli(examples: Sequence[Union[tf.train.Example,
output_path: Output path for the dataset. output_path: Output path for the dataset.
""" """
with io.FileIO(output_path, 'wb') as fileio: with io.FileIO(output_path, 'wb') as fileio:
import riegeli # pylint: disable=g-import-not-at-top
with riegeli.RecordWriter(fileio) as writer: with riegeli.RecordWriter(fileio) as writer:
writer.write_messages(examples) writer.write_messages(examples)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment