File tree Expand file tree Collapse file tree 1 file changed +7
-1
lines changed
tensorflow_datasets/core/dataset_builders Expand file tree Collapse file tree 1 file changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -198,6 +198,7 @@ def __init__(
198
198
hf_num_proc : Optional [int ] = None ,
199
199
tfds_num_proc : Optional [int ] = None ,
200
200
ignore_hf_errors : bool = False ,
201
+ overwrite_version : str | None = None ,
201
202
** config_kwargs ,
202
203
):
203
204
self ._hf_repo_id = hf_repo_id
@@ -216,7 +217,12 @@ def __init__(
216
217
f' hf_repo_id={ self ._hf_repo_id } , hf_config={ self ._hf_config } ,'
217
218
f' config_kwargs={ self .config_kwargs } '
218
219
) 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
+ )
220
226
self .VERSION = version_lib .Version (version ) # pylint: disable=invalid-name
221
227
if self ._hf_config :
222
228
self ._converted_builder_config = dataset_builder .BuilderConfig (
You can’t perform that action at this time.
0 commit comments