Fix OmDet test after arg deprecation (#39766)

fix arg name
This commit is contained in:
Cyril Vallez
2025-07-29 22:10:36 +02:00
committed by GitHub
parent 4abb053b6c
commit ddd2100767

View File

@@ -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(