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() ) == [