chore: Fix flaky evals test

sararob · copybara-github · commit 8385dd51de6c · 2026-05-20T10:26:02.000-07:00
PiperOrigin-RevId: 918518660
diff --git a/tests/unit/vertexai/test_evaluation.py b/tests/unit/vertexai/test_evaluation.py
@@ -1074,8 +1074,8 @@ 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()
+        assert sorted(
+            list(test_result.metrics_table["summarization_quality/score"].to_list())
         ) == [5, 4]
         assert list(
             test_result.metrics_table["summarization_quality/explanation"].to_list()

Original file line number	Diff line number	Diff line change
`@@ -1074,8 +1074,8 @@ 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()`
	`1077`	`+ assert sorted(`
	`1078`	`+ list(test_result.metrics_table["summarization_quality/score"].to_list())`
`1079`	`1079`	`) == [5, 4]`
`1080`	`1080`	`assert list(`
`1081`	`1081`	`test_result.metrics_table["summarization_quality/explanation"].to_list()`