fix make_image (#7768)

8071c177 · Philip Meier · GitHub · b9b7cfc6 · 8071c177
Unverified Commit 8071c177 authored Jul 27, 2023 by Philip Meier Committed by GitHub Jul 27, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

test/common_utils.py test/common_utils.py +2 -1

No files found.
--- a/test/common_utils.py
+++ b/test/common_utils.py
@@ -503,12 +503,13 @@ def make_image(
    device="cpu",
    memory_format=torch.contiguous_format,
 ):
+    dtype = dtype or torch.uint8
    max_value = get_max_value(dtype)
    data = torch.testing.make_tensor(
        (*batch_dims, get_num_channels(color_space), *size),
        low=0,
        high=max_value,
-        dtype=dtype or torch.uint8,
+        dtype=dtype,
        device=device,
        memory_format=memory_format,
    )