Skip to content

Commit ddd2100

Browse files
authored
Fix OmDet test after arg deprecation (#39766)
fix arg name
1 parent 4abb053 commit ddd2100

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/models/omdet_turbo/test_modeling_omdet_turbo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -844,7 +844,7 @@ def test_inference_object_detection_head_batched(self):
844844
outputs,
845845
text_labels=text_labels_batched,
846846
target_sizes=[image.size[::-1] for image in images_batched],
847-
score_threshold=0.2,
847+
threshold=0.2,
848848
)
849849
expected_scores = torch.tensor([0.7675, 0.3016, 0.7454]).to(torch_device)
850850
expected_slice_boxes = torch.tensor(

0 commit comments

Comments
 (0)