diff --git a/tensorflow_datasets/core/dataset_builder.py b/tensorflow_datasets/core/dataset_builder.py
index 5e930184dd9..c4d0befd769 100644
--- a/tensorflow_datasets/core/dataset_builder.py
+++ b/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()
 
 
diff --git a/tensorflow_datasets/core/dataset_builder_test.py b/tensorflow_datasets/core/dataset_builder_test.py
index 9169cf3b7fd..a0be516c3a4 100644
--- a/tensorflow_datasets/core/dataset_builder_test.py
+++ b/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