Fix huggingface_utils.to_huggingface_name.

PiperOrigin-RevId: 704654760

Author: fineguy

tensorflow_datasets/core/utils/huggingface_utils.py

@@ -139,12 +139,12 @@ def to_huggingface_name(tfds_dataset_name: str) -> str:
     DatasetNotFoundError: If the TFDS dataset name doesn't correspond to any
     existing Huggingface dataset.
   """
-  for hf_dataset_name in huggingface_hub.list_datasets():
+  for hf_dataset_info in huggingface_hub.list_datasets():
     if (
-        conversion_utils.to_tfds_name(hf_dataset_name)
+        conversion_utils.to_tfds_name(hf_dataset_info.id)
         == tfds_dataset_name.lower()
     ):
-      return hf_dataset_name
+      return hf_dataset_info.id
   raise registered.DatasetNotFoundError(
       f'"{tfds_dataset_name}" is not listed in Huggingface datasets.'
   )

tensorflow_datasets/core/utils/huggingface_utils_test.py

@@ -113,7 +113,10 @@ def test_convert_hf_features(hf_features, tfds_features):
 @pytest.fixture(name='mock_list_datasets')
 def _list_datasets(monkeypatch):
   def mock_list_datasets():
-    return ['mnist', 'bigscience/P3', 'x', 'x/Y-z', 'fashion_mnist']
+    return [
+        huggingface_hub.DatasetInfo(id=name)
+        for name in ('mnist', 'bigscience/P3', 'x', 'x/Y-z', 'fashion_mnist')
+    ]
 
   monkeypatch.setattr(huggingface_hub, 'list_datasets', mock_list_datasets)