[fbsync] fix make_image (#7768)

Reviewed By: matteobettini

Differential Revision: D48642304

fbshipit-source-id: 103fba520209fd0fc81ecc839e56cdea087b7981

Author: NicolasHug

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,
     )