This commit is contained in:
yaswant19 2025-07-01 23:58:48 +05:30
parent 7e5c71e8d2
commit eb44529438

View File

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