Automated Code Change

PiperOrigin-RevId: 753088107

Author: h-joo

tensorflow_datasets/core/dataset_info.py

@@ -419,7 +419,7 @@ def set_nondeterministic_order(self, nondeterministic_order: bool) -> None:
   def homepage(self) -> str:
     urls = self.as_proto.location.urls
     tfds_homepage = f"https://www.tensorflow.org/datasets/catalog/{self.name}"
-    return urls and urls[0] or tfds_homepage
+    return urls and urls[0] or tfds_homepage  # pytype: disable=bad-return-type
 
   @property
   def citation(self) -> str:

tensorflow_datasets/core/load.py

@@ -379,7 +379,7 @@ def load_dataset(
       raise RuntimeError(
           f'Unsupported return type {type(load_output)} of `load` function.'
       )
-    return loaded_datasets
+    return loaded_datasets  # pytype: disable=bad-return-type
 
   def load_datasets(
       self,