Internal change

PiperOrigin-RevId: 704254050
tensorflow · Dec 9, 2024 · 5d2c3e5 · 5d2c3e5
1 parent 4aa203e
commit 5d2c3e5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tensorflow_datasets/core/utils/shard_utils.py b/tensorflow_datasets/core/utils/shard_utils.py
@@ -20,9 +20,10 @@
  - tfrecord_writer, to read sharded bucket files (temp files), based on final
  sharding needs.
 """
+
 from __future__ import annotations
 
-from collections.abc import Sequence
+from collections.abc import Iterable, Sequence
 import dataclasses
 import math
 import os

diff --git a/tensorflow_datasets/core/utils/shard_utils_test.py b/tensorflow_datasets/core/utils/shard_utils_test.py
@@ -240,6 +240,5 @@ def test_incorrect_values(self, filename, skip, take, examples_in_shard, msg):
     with self.assertRaisesRegex(ValueError, expected_regex=msg):
       shard_utils.FileInstruction(filename, skip, take, examples_in_shard)
 
-
 if __name__ == '__main__':
   testing.test_main()