Skip to content

Commit 83974a8

Browse files
tomvdwThe TensorFlow Datasets Authors
authored andcommitted
Handle case when builder config name contains the version
PiperOrigin-RevId: 695338267
1 parent 498b807 commit 83974a8

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tensorflow_datasets/core/dataset_builder.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1359,6 +1359,8 @@ def _create_builder_config(
13591359
return None
13601360

13611361
if isinstance(builder_config, str):
1362+
if version is None and ":" in builder_config:
1363+
builder_config, version = builder_config.split(":")
13621364
config = self.get_builder_config(name=builder_config, version=version)
13631365
if config is not None:
13641366
return config

0 commit comments

Comments
 (0)