From ddd2100767dfffd0e342cca2422fb7da8c00aea3 Mon Sep 17 00:00:00 2001 From: Cyril Vallez Date: Tue, 29 Jul 2025 22:10:36 +0200 Subject: [PATCH] Fix OmDet test after arg deprecation (#39766) fix arg name --- tests/models/omdet_turbo/test_modeling_omdet_turbo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/omdet_turbo/test_modeling_omdet_turbo.py b/tests/models/omdet_turbo/test_modeling_omdet_turbo.py index 9d76ad392c..b7119d7973 100644 --- a/tests/models/omdet_turbo/test_modeling_omdet_turbo.py +++ b/tests/models/omdet_turbo/test_modeling_omdet_turbo.py @@ -844,7 +844,7 @@ class OmDetTurboModelIntegrationTests(unittest.TestCase): outputs, text_labels=text_labels_batched, target_sizes=[image.size[::-1] for image in images_batched], - score_threshold=0.2, + threshold=0.2, ) expected_scores = torch.tensor([0.7675, 0.3016, 0.7454]).to(torch_device) expected_slice_boxes = torch.tensor(