Remove dataset argument from list_data_dirs and get_default_data_dir.

PiperOrigin-RevId: 678588984

Author: fineguy

tensorflow_datasets/core/dataset_builder.py

@@ -1162,9 +1162,7 @@ def _build_data_dir(self, given_data_dir: Optional[str]):
     default_data_dir = file_utils.get_default_data_dir(
         given_data_dir=given_data_dir
     )
-    all_data_dirs = file_utils.list_data_dirs(
-        given_data_dir=given_data_dir, dataset=self.name
-    )
+    all_data_dirs = file_utils.list_data_dirs(given_data_dir=given_data_dir)
 
     all_versions = set()
     requested_version_dirs = {}

tensorflow_datasets/core/utils/file_utils.py

@@ -128,15 +128,13 @@ def incomplete_dir(
 
 def list_data_dirs(
     given_data_dir: ListOrElem[PathLike] | None = None,
-    dataset: str | None = None,
 ) -> Sequence[PathLike]:
   """Return the list of all `data_dir` to look-up.
 
   Args:
     given_data_dir: If a `data_dir` is provided, only the explicitly given
       `data_dir` will be returned, otherwise the list of all registered data_dir
       is returned
-    dataset: Dataset to load.
 
   Returns:
     The list of all data_dirs to look-up.
@@ -148,23 +146,18 @@ def list_data_dirs(
     else:
       return [given_data_dir]
   else:
-    default_data_dir = get_default_data_dir(
-        given_data_dir=given_data_dir, dataset=dataset
-    )
+    default_data_dir = get_default_data_dir(given_data_dir=given_data_dir)
     all_data_dirs = _registered_data_dir | {default_data_dir}
     return sorted(os.path.expanduser(d) for d in all_data_dirs)
 
 
-def get_default_data_dir(
-    given_data_dir: str | None = None, dataset: str | None = None
-) -> str:
+def get_default_data_dir(given_data_dir: str | None = None) -> str:
   """Returns the default data_dir."""
   if given_data_dir:
     return os.path.expanduser(given_data_dir)
   elif 'TFDS_DATA_DIR' in os.environ:
     return os.environ['TFDS_DATA_DIR']
   else:
-    del dataset
     return constants.DATA_DIR
 
 

tensorflow_datasets/core/utils/file_utils_test.py

@@ -31,7 +31,6 @@
 def test_default_data_dir():
   data_dir = file_utils.get_default_data_dir(given_data_dir=None)
   assert data_dir
-  assert isinstance(data_dir, str)
 
 
 def test_list_dataset_variants_with_configs(mock_fs: testing.MockFs):