Skip to content

Commit fc11f4c

Browse files
author
The TensorFlow Datasets Authors
committed
Prevent load_read_only_builder from modifying input builder_kwargs.
PiperOrigin-RevId: 798187843
1 parent d92e86e commit fc11f4c

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tensorflow_datasets/core/read_only_builder.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -314,18 +314,19 @@ def builder_from_files(
314314
DatasetNotFoundError: If the dataset cannot be loaded.
315315
"""
316316
# Find and load dataset builder.
317-
builder_dir = _find_builder_dir(name, **builder_kwargs)
317+
copy_builder_kwargs = dict(builder_kwargs)
318+
builder_dir = _find_builder_dir(name, **copy_builder_kwargs)
318319
if builder_dir is None:
319320
data_dirs = file_utils.list_data_dirs(
320-
given_data_dir=builder_kwargs.get('data_dir')
321+
given_data_dir=copy_builder_kwargs.get('data_dir')
321322
)
322323
raise registered.DatasetNotFoundError(
323324
f'Could not find dataset files for: {name}. Make sure you have the'
324325
' correct permissions to access the dataset '
325326
f'and that it has been generated in: {data_dirs}. If the dataset has'
326327
' configs, you might have to specify the config name.'
327328
)
328-
file_format = builder_kwargs.pop('file_format', None)
329+
file_format = copy_builder_kwargs.pop('file_format', None)
329330
return builder_from_directory(builder_dir, file_format=file_format)
330331

331332

0 commit comments

Comments
 (0)