Move the logic to read the dataset info from the constructor to the info property.

PiperOrigin-RevId: 681031052

Author: fineguy

tensorflow_datasets/core/dataset_builder.py

@@ -289,6 +289,23 @@ def __init__(
     self._version = self._pick_version(version)
     # Compute the base directory (for download) and dataset/version directory.
     self._data_dir_root, self._data_dir = self._build_data_dir(data_dir)
+    # If the dataset info is available, use it.
+    if dataset_info.dataset_info_path(self.data_path).exists():
+      self.info.read_from_directory(self._data_dir)
+    else:  # Use the code version (do not restore data)
+      self.info.initialize_from_bucket()
+    if self.BLOCKED_VERSIONS is not None:
+      config_name = self._builder_config.name if self._builder_config else None
+      if is_blocked := self.BLOCKED_VERSIONS.is_blocked(
+          version=self._version, config=config_name
+      ):
+        default_msg = (
+            f"Dataset {self.name} is blocked at version {self._version} and"
+            f" config {config_name}."
+        )
+        self.info.set_is_blocked(
+            is_blocked.blocked_msg if is_blocked.blocked_msg else default_msg
+        )
 
   @utils.classproperty
   @classmethod
@@ -493,23 +510,6 @@ def info(self) -> dataset_info.DatasetInfo:
           "DatasetBuilder._info should returns `tfds.core.DatasetInfo`, not "
           f" {type(info)}."
       )
-    # If the dataset info is available, use it.
-    if dataset_info.dataset_info_path(self.data_path).exists():
-      info.read_from_directory(self.data_path)
-    else:  # Use the code version (do not restore data)
-      info.initialize_from_bucket()
-    if self.BLOCKED_VERSIONS is not None:
-      config_name = self._builder_config.name if self._builder_config else None
-      if is_blocked := self.BLOCKED_VERSIONS.is_blocked(
-          version=self._version, config=config_name
-      ):
-        default_msg = (
-            f"Dataset {self.name} is blocked at version {self._version} and"
-            f" config {config_name}."
-        )
-        info.set_is_blocked(
-            is_blocked.blocked_msg if is_blocked.blocked_msg else default_msg
-        )
     return info
 
   @utils.classproperty

tensorflow_datasets/core/dataset_info.py

@@ -707,7 +707,7 @@ def read_from_directory(self, dataset_info_dir: epath.PathLike) -> None:
       )
 
     self._identity = DatasetIdentity.from_proto(
-        info_proto=parsed_proto, data_dir=os.fspath(dataset_info_dir)
+        info_proto=parsed_proto, data_dir=dataset_info_dir
     )
 
     # Update splits