chore: Fix flaky evals test

sararob · copybara-github · commit adfc80077a4d · 2026-05-20T10:38:39.000-07:00
PiperOrigin-RevId: 918518660
diff --git a/tests/unit/vertexai/test_evaluation.py b/tests/unit/vertexai/test_evaluation.py
@@ -1074,9 +1074,9 @@ def test_compute_pointwise_metrics_metric_prompt_template_example_string_model(
                 "summarization_quality/explanation",
             ]
         )
-        assert list(
-            test_result.metrics_table["summarization_quality/score"].to_list()
-        ) == [5, 4]
+        assert sorted(
+            list(test_result.metrics_table["summarization_quality/score"].to_list())
+        ) == [4, 5]
         assert list(
             test_result.metrics_table["summarization_quality/explanation"].to_list()
         ) == [

Original file line number	Diff line number	Diff line change
`@@ -1074,9 +1074,9 @@ def test_compute_pointwise_metrics_metric_prompt_template_example_string_model(`
`1074`	`1074`	`"summarization_quality/explanation",`
`1075`	`1075`	`]`
`1076`	`1076`	`)`
`1077`		`- assert list(`
`1078`		`- test_result.metrics_table["summarization_quality/score"].to_list()`
`1079`		`- ) == [5, 4]`
	`1077`	`+ assert sorted(`
	`1078`	`+ list(test_result.metrics_table["summarization_quality/score"].to_list())`
	`1079`	`+ ) == [4, 5]`
`1080`	`1080`	`assert list(`
`1081`	`1081`	`test_result.metrics_table["summarization_quality/explanation"].to_list()`
`1082`	`1082`	`) == [`