Skip to content

Commit e5b013e

Browse files
Alfonso CastañoThe TensorFlow Datasets Authors
authored andcommitted
Add overwrite_version to HuggingFaceDatasetBuilder
PiperOrigin-RevId: 642897287
1 parent 2cd3d44 commit e5b013e

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

tensorflow_datasets/core/dataset_builders/huggingface_dataset_builder.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,6 +198,7 @@ def __init__(
198198
hf_num_proc: Optional[int] = None,
199199
tfds_num_proc: Optional[int] = None,
200200
ignore_hf_errors: bool = False,
201+
overwrite_version: str | None = None,
201202
**config_kwargs,
202203
):
203204
self._hf_repo_id = hf_repo_id
@@ -216,7 +217,12 @@ def __init__(
216217
f' hf_repo_id={self._hf_repo_id}, hf_config={self._hf_config},'
217218
f' config_kwargs={self.config_kwargs}'
218219
) from e
219-
version = str(self._hf_info.version or self._hf_builder.VERSION or '1.0.0')
220+
version = str(
221+
overwrite_version
222+
or self._hf_info.version
223+
or self._hf_builder.VERSION
224+
or '1.0.0'
225+
)
220226
self.VERSION = version_lib.Version(version) # pylint: disable=invalid-name
221227
if self._hf_config:
222228
self._converted_builder_config = dataset_builder.BuilderConfig(

0 commit comments

Comments
 (0)