From 870edfdf65692077a655a18a3b3f7b2d47177652 Mon Sep 17 00:00:00 2001 From: Philip Meier Date: Thu, 25 Feb 2021 11:39:35 +0100 Subject: [PATCH] [skip CI] fix kwargs forwarding in fake data utility functions (#3459) --- test/datasets_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/datasets_utils.py b/test/datasets_utils.py index 337a7382366..4e3fd0ac0e3 100644 --- a/test/datasets_utils.py +++ b/test/datasets_utils.py @@ -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 size(idx): 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) ]