Add a hook to be executed after download and prepare.

This can be used to execute custom logic after download_and_prepare, such as post-processing the data or generating additional files.

PiperOrigin-RevId: 645042072

Author: tomvdw

tensorflow_datasets/core/dataset_builder.py

@@ -731,6 +731,17 @@ def download_and_prepare(
 
     self._log_download_done()
 
+    # Execute post download and prepare hook if it exists.
+    self._post_download_and_prepare_hook()
+
+
+  def _post_download_and_prepare_hook(self) -> None:
+    """Hook to be executed after download and prepare.
+
+    Override this in custom dataset builders to execute custom logic after
+    download and prepare.
+    """
+    pass
 
   def _update_dataset_info(self) -> None:
     """Updates the `dataset_info.json` file in the dataset dir."""