Internal change.

PiperOrigin-RevId: 590462655

seqio/experimental.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 """Experimental utilities for SeqIO."""
+
 import functools
 import inspect
 from typing import Callable, Iterable, Mapping, Optional, Sequence
@@ -86,6 +87,7 @@ def _no_op_mixture_registry_get(*args, **kwargs):
 
 
 def disable_registry():
+  """Disables the seqio TaskRegistry and MixtureRegistry."""
   _enfore_empty_registries()
   dataset_providers.TaskRegistry.add = _no_op_task_registry_add
   dataset_providers.TaskRegistry.add_provider = _no_op_task_registry_add

seqio/experimental_test.py

@@ -15,6 +15,8 @@
 """Tests for seqio.preprocessors."""
 
 import contextlib
+from unittest import mock
+
 from absl.testing import absltest
 from seqio import dataset_providers
 from seqio import experimental
@@ -984,5 +986,6 @@ def test_mixture_registry_get_error(self):
         MixtureRegistry.get('dummy_mixture')
 
 
+
 if __name__ == '__main__':
   absltest.main()