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

pytorch · Feb 25, 2021 · 870edfd · 870edfd
1 parent 13c4470
commit 870edfd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)
     ]