Skip to content

GitLab

Unverified Commit feec5695 authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix Detr CI (#25972) 



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 404ff8fc
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/detr/test_modeling_detr.py
View file @ feec5695
...@@ -606,7 +606,7 @@ class DetrModelIntegrationTestsTimmBackbone(unittest.TestCase): ...@@ -606,7 +606,7 @@ class DetrModelIntegrationTestsTimmBackbone(unittest.TestCase):
torch_device torch_device
) )
expected_number_of_segments = 5 expected_number_of_segments = 5
expected_first_segment = {"id": 1, "label_id": 17, "was_fused": False, "score": 0.994097} expected_first_segment = {"id": 1, "label_id": 17, "was_fused": False, "score": 0.994096}
number_of_unique_segments = len(torch.unique(results["segmentation"])) number_of_unique_segments = len(torch.unique(results["segmentation"]))
self.assertTrue( self.assertTrue(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment