fix oin in test

pytorch · Oct 18, 2023 · 9ad5f81 · 9ad5f81
1 parent 7825de3
commit 9ad5f81
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_utils.py b/test/test_utils.py
@@ -105,7 +105,7 @@ def test_draw_boxes():
         res = Image.fromarray(result.permute(1, 2, 0).contiguous().numpy())
         res.save(path)
 
-    if PILLOW_VERSION >= (8, 2):
+    if PILLOW_VERSION >= (10, 1):
         # The reference image is only valid for new PIL versions
         expected = torch.as_tensor(np.array(Image.open(path))).permute(2, 0, 1)
         assert_equal(result, expected)