SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix OmDet test after arg deprecation (#39766)

Browse Source 
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
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/omdet_turbo/test_modeling_omdet_turbo.py
View File

@@ -844,7 +844,7 @@ class OmDetTurboModelIntegrationTests(unittest.TestCase):
outputs, outputs,
text_labels=text_labels_batched, text_labels=text_labels_batched,
target_sizes=[image.size[::-1] for image in images_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_scores = torch.tensor([0.7675, 0.3016, 0.7454]).to(torch_device)
expected_slice_boxes = torch.tensor( expected_slice_boxes = torch.tensor(