From 0ec70b0054c1b5961522b1adc2df8d0c68545f74 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 23 Jul 2024 19:33:15 +0200 Subject: [PATCH] Windows `torch==2.4.0` Segment `augment=True` failed test bypass (#14637) Signed-off-by: Glenn Jocher Co-authored-by: UltralyticsAssistant --- tests/test_python.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_python.py b/tests/test_python.py index 97441a78..062716dd 100644 --- a/tests/test_python.py +++ b/tests/test_python.py @@ -95,7 +95,7 @@ def test_predict_img(model_name): Image.open(SOURCE), # PIL np.zeros((320, 640, 3), dtype=np.uint8), # numpy ] - assert len(model(batch, imgsz=32, augment=True)) == len(batch) # multiple sources in a batch + assert len(model(batch, imgsz=32)) == len(batch) # multiple sources in a batch @pytest.mark.parametrize("model", MODELS)