@@ -553,14 +553,10 @@ def get_reference(
553
553
Returns:
554
554
a reference to this instantiated builder.
555
555
"""
556
- if self .builder_config :
557
- config = self .builder_config .name
558
- else :
559
- config = None
560
556
return naming .DatasetReference (
561
557
dataset_name = self .name ,
562
558
namespace = namespace ,
563
- config = config ,
559
+ config = self . builder_config_name ,
564
560
version = self .version ,
565
561
data_dir = self .data_dir_root ,
566
562
)
@@ -576,19 +572,17 @@ def is_prepared(self) -> bool:
576
572
577
573
def is_blocked (self ) -> utils .IsBlocked :
578
574
"""Returns whether this builder (version, config) is blocked."""
579
- config_name = self .builder_config .name if self .builder_config else None
580
575
if blocked_versions := self .blocked_versions :
581
576
return blocked_versions .is_blocked (
582
- version = self .version , config = config_name
577
+ version = self .version , config = self . builder_config_name
583
578
)
584
579
return utils .IsBlocked (False )
585
580
586
581
def assert_is_not_blocked (self ) -> None :
587
582
"""Checks that the dataset is not blocked."""
588
- config_name = self .builder_config .name if self .builder_config else None
589
583
if blocked_versions := self .blocked_versions :
590
584
is_blocked = blocked_versions .is_blocked (
591
- version = self .version , config = config_name
585
+ version = self .version , config = self . builder_config_name
592
586
)
593
587
if is_blocked .result :
594
588
raise utils .DatasetVariantBlockedError (is_blocked .blocked_msg )
@@ -1382,6 +1376,11 @@ def builder_config(self) -> Optional[Any]:
1382
1376
"""`tfds.core.BuilderConfig` for this builder."""
1383
1377
return self ._builder_config
1384
1378
1379
+ @property
1380
+ def builder_config_name (self ) -> str | None :
1381
+ """Name of the `tfds.core.BuilderConfig` for this builder."""
1382
+ return self ._builder_config .name if self ._builder_config else None
1383
+
1385
1384
def _create_builder_config (
1386
1385
self ,
1387
1386
builder_config : str | BuilderConfig | None ,
0 commit comments