Skip to content

Commit 4fd6d59

Browse files
tomvdwThe TensorFlow Datasets Authors
authored andcommitted
Do a more precise check whether a dataset already exists
If the dataset folder exists, but is empty, then `read_from_directory` fails. PiperOrigin-RevId: 649046864
1 parent 7247953 commit 4fd6d59

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tensorflow_datasets/core/dataset_builder.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -286,7 +286,8 @@ def __init__(
286286
self._version = self._pick_version(version)
287287
# Compute the base directory (for download) and dataset/version directory.
288288
self._data_dir_root, self._data_dir = self._build_data_dir(data_dir)
289-
if self.data_path.exists():
289+
# If the dataset info is available, use it.
290+
if dataset_info.dataset_info_path(self.data_path).exists():
290291
self.info.read_from_directory(self._data_dir)
291292
else: # Use the code version (do not restore data)
292293
self.info.initialize_from_bucket()

0 commit comments

Comments
 (0)