Internal change

PiperOrigin-RevId: 788353351

Author: The TensorFlow Datasets Authors

tensorflow_datasets/core/load.py

@@ -49,6 +49,7 @@
 from tensorflow_datasets.core.utils import version as version_lib
 from tensorflow_datasets.core.utils.lazy_imports_utils import tensorflow as tf
 
+
 # pylint: disable=logging-format-interpolation
 
 Tree = type_utils.Tree
@@ -183,6 +184,7 @@ def get_dataset_repr() -> str:
           f' explicitly set. Wrong arguments for {get_dataset_repr()}'
       )
     builder_kwargs['data_dir'] = gcs_utils.gcs_path('datasets')
+
   if name.namespace:
     if name.namespace == 'huggingface':
       return huggingface_dataset_builder.builder(
@@ -827,12 +829,8 @@ def data_source(
   """  # fmt:skip
   builder_kwargs = _set_file_format_for_data_source(data_dir, builder_kwargs)
   _validate_file_format_for_data_source(builder_kwargs)
-  dbuilder = _fetch_builder(
-      name,
-      data_dir,
-      builder_kwargs,
-      try_gcs,
-  )
+
+  dbuilder = _fetch_builder(name, data_dir, builder_kwargs, try_gcs)
   _download_and_prepare_builder(dbuilder, download, download_and_prepare_kwargs)
   return dbuilder.as_data_source(
       split=split, decoders=decoders, deserialize_method=deserialize_method

tensorflow_datasets/testing/dataset_builder_testing.py

@@ -425,7 +425,10 @@ def _test_checksums(self):
         f"Some urls checksums are missing at: {filepath}\n{err_msg}",
     )
 
-  def _download_and_prepare_as_dataset(self, builder):
+  def _download_and_prepare_as_dataset(
+      self,
+      builder,
+  ):
     # Provide the manual dir only if builder has MANUAL_DOWNLOAD_INSTRUCTIONS
     # set.
 
@@ -461,7 +464,9 @@ def _download_and_prepare_as_dataset(self, builder):
           beam_runner=beam_runner,
       )
       with self._test_key_not_local_path(builder):
-        builder.download_and_prepare(download_config=download_config)
+        builder.download_and_prepare(
+            download_config=download_config,
+        )
 
     with self._subTest("as_dataset"):
       self._assertAsDataset(builder)