Skip to content

Commit ff4e90c

Browse files
Conchylicultorcopybara-github
authored andcommitted
Fix wrong auto-cache documentation value when splits info are missing
PiperOrigin-RevId: 296294659
1 parent 643ee98 commit ff4e90c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tensorflow_datasets/scripts/templates/dataset.mako.md

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,9 @@ def build_autocached_info(builder):
8282
unshuffle_cached.add(split_name)
8383

8484

85-
if len(always_cached) == len(builder.info.splits.keys()):
85+
if not len(builder.info.splits):
86+
autocached_info = 'Unknown'
87+
elif len(always_cached) == len(builder.info.splits.keys()):
8688
autocached_info = 'Yes' # All splits are auto-cached.
8789
elif len(never_cached) == len(builder.info.splits.keys()):
8890
autocached_info = 'No' # Splits never auto-cached.

0 commit comments

Comments
 (0)