Unverified Commit 870edfdf authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

[skip CI] fix kwargs forwarding in fake data utility functions (#3459)

parent 13c4470a
......@@ -572,7 +572,7 @@ def create_image_file(
image = create_image_or_video_tensor(size)
file = pathlib.Path(root) / name
PIL.Image.fromarray(image.permute(2, 1, 0).numpy()).save(file)
PIL.Image.fromarray(image.permute(2, 1, 0).numpy()).save(file, **kwargs)
return file
......@@ -708,6 +708,6 @@ def create_video_folder(
os.makedirs(root)
return [
create_video_file(root, file_name_fn(idx), size=size(idx) if callable(size) else size)
create_video_file(root, file_name_fn(idx), size=size(idx) if callable(size) else size, **kwargs)
for idx in range(num_examples)
]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment