Skip to content

GitLab

Commit 64ecc937 authored by Jiageng Zhang's avatar Jiageng Zhang Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 475446008
parent f20a9ef6
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
official/core/file_writers_test.py
View file @ 64ecc937
...@@ -31,7 +31,7 @@ class FileWritersTest(tf.test.TestCase, parameterized.TestCase): ...@@ -31,7 +31,7 @@ class FileWritersTest(tf.test.TestCase, parameterized.TestCase):
self._example = example_builder.example self._example = example_builder.example
@parameterized.parameters('tfrecord', 'TFRecord', 'tfrecord_compressed', @parameterized.parameters('tfrecord', 'TFRecord', 'tfrecord_compressed',
'TFRecord_Compressed', 'riegeli', 'Riegeli') 'TFRecord_Compressed')
def test_write_small_dataset_success(self, file_type): def test_write_small_dataset_success(self, file_type):
temp_dir = self.create_tempdir() temp_dir = self.create_tempdir()
temp_dataset_file = os.path.join(temp_dir.full_path, 'train') temp_dataset_file = os.path.join(temp_dir.full_path, 'train')
......
official/requirements.txt
View file @ 64ecc937
...@@ -14,7 +14,6 @@ tensorflow-addons ...@@ -14,7 +14,6 @@ tensorflow-addons
dataclasses;python_version<"3.7" dataclasses;python_version<"3.7"
gin-config gin-config
tf_slim>=1.1.0 tf_slim>=1.1.0
riegeli
Cython Cython
matplotlib matplotlib
# Loader becomes a required positional argument in 6.0 in yaml.load # Loader becomes a required positional argument in 6.0 in yaml.load
......
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