From ced40e4db624cc66515c9e24472c8abd0c7d564f Mon Sep 17 00:00:00 2001 From: yaswant19 Date: Wed, 2 Jul 2025 15:05:18 +0530 Subject: [PATCH] expected values update --- tests/models/eomt/test_modeling_eomt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/eomt/test_modeling_eomt.py b/tests/models/eomt/test_modeling_eomt.py index dfcae99c736..c4b026cc18e 100644 --- a/tests/models/eomt/test_modeling_eomt.py +++ b/tests/models/eomt/test_modeling_eomt.py @@ -476,7 +476,7 @@ class EomtForUniversalSegmentationIntegrationTest(unittest.TestCase): pipe = pipeline(model=self.model_id, subtask="panoptic", device=torch_device) output = pipe(image) - EXPECTED_OUTPUT_LABELS = ["LABEL_15", "LABEL_15", "LABEL_57", "LABEL_65", "LABEL_65"] + EXPECTED_OUTPUT_LABELS = ["cat", "cat", "couch", "remote", "remote"] output_labels = [segment["label"] for segment in output] self.assertEqual(output_labels, EXPECTED_OUTPUT_LABELS)