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( ...@@ -572,7 +572,7 @@ def create_image_file(
image = create_image_or_video_tensor(size) image = create_image_or_video_tensor(size)
file = pathlib.Path(root) / name 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 return file
...@@ -708,6 +708,6 @@ def create_video_folder( ...@@ -708,6 +708,6 @@ def create_video_folder(
os.makedirs(root) os.makedirs(root)
return [ 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) 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