Include trust_remote_code flag for code-based HF builders preparation.

PiperOrigin-RevId: 661246526

Author: The TensorFlow Datasets Authors

tensorflow_datasets/core/dataset_builders/huggingface_dataset_builder.py

@@ -200,6 +200,7 @@ def __init__(
       tfds_num_proc: Optional[int] = None,
       ignore_hf_errors: bool = False,
       overwrite_version: str | None = None,
+      trust_remote_code: bool = False,
       **config_kwargs,
   ):
     self._hf_repo_id = hf_repo_id
@@ -210,7 +211,10 @@ def __init__(
     )
     try:
       self._hf_builder = hf_datasets.load_dataset_builder(
-          self._hf_repo_id, self._hf_config, **self.config_kwargs
+          self._hf_repo_id,
+          self._hf_config,
+          trust_remote_code=trust_remote_code,
+          **self.config_kwargs,
       )
     except Exception as e:
       raise RuntimeError(

tensorflow_datasets/core/dataset_builders/huggingface_dataset_builder_test.py

@@ -104,7 +104,10 @@ def mock_huggingface_dataset_builder(
       other_arg='this is another arg',
   )
   load_dataset_builder.assert_called_once_with(
-      'foo/bar', 'config', other_arg='this is another arg'
+      'foo/bar',
+      'config',
+      other_arg='this is another arg',
+      trust_remote_code=False,
   )
   login_to_hf.assert_called_once_with('SECRET_TOKEN')
   yield builder