lint

pytorch · Apr 3, 2024 · ccdc070 · ccdc070
1 parent d3d2f79
commit ccdc070
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_transforms_v2.py b/test/test_transforms_v2.py
@@ -5330,7 +5330,7 @@ def test_functional_error(self):
 @pytest.mark.parametrize("f", [F.to_tensor, F.pil_to_tensor])
 def test_I16_to_tensor(f):
     # See https://github.com/pytorch/vision/issues/8359
-    I16_pil_img = PIL.Image.fromarray(np.random.randint(0, 2 ** 16, (10, 10), dtype=np.uint16))
+    I16_pil_img = PIL.Image.fromarray(np.random.randint(0, 2**16, (10, 10), dtype=np.uint16))
     assert I16_pil_img.mode == "I;16"
 
     cm = pytest.warns(UserWarning, match="deprecated") if f is F.to_tensor else contextlib.nullcontext()