From 75684a64dde6ae288ae47fe3f0bb722897ce140f Mon Sep 17 00:00:00 2001 From: Sara Robinson Date: Wed, 20 May 2026 11:13:39 -0700 Subject: [PATCH] chore: Fix flaky evals test PiperOrigin-RevId: 918549002 --- tests/unit/vertexai/test_evaluation.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/unit/vertexai/test_evaluation.py b/tests/unit/vertexai/test_evaluation.py index 627d22a01d..8f37cf1ca3 100644 --- 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() ) == [