Internal change.

PiperOrigin-RevId: 779158253

Author: fineguy

tensorflow_datasets/core/dataset_builder.py

@@ -620,7 +620,7 @@ def download_and_prepare(
       RuntimeError: when the config cannot be found.
       DatasetBlockedError: if the given version, or combination of version and
         config, has been marked as blocked in the builder's BLOCKED_VERSIONS.
-    """
+    """  # fmt: skip
     self.assert_is_not_blocked()
 
 

tensorflow_datasets/core/dataset_builder_test.py

@@ -20,7 +20,6 @@
 import tempfile
 from unittest import mock
 
-from absl.testing import flagsaver
 from absl.testing import parameterized
 import dill
 from etils import epath

tensorflow_datasets/scripts/cli/cli_utils.py

@@ -20,6 +20,7 @@
 import itertools
 import os
 import pathlib
+from typing import cast
 
 from absl import logging
 from etils import epath