mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-03 21:00:08 +06:00
🙈 MB
This commit is contained in:
parent
7e5c71e8d2
commit
eb44529438
@ -478,5 +478,5 @@ class EomtForUniversalSegmentationIntegrationTest(unittest.TestCase):
|
||||
|
||||
EXPECTED_OUTPUT_LABELS = ["LABEL_15", "LABEL_15", "LABEL_57", "LABEL_65", "LABEL_65"]
|
||||
|
||||
output_labels = [segment["label"] for segment in output["segments_info"]]
|
||||
output_labels = [segment["label"] for segment in output]
|
||||
self.assertEqual(output_labels, EXPECTED_OUTPUT_LABELS)
|
||||
|
Loading…
Reference in New Issue
Block a user