From 26342d181798e4e21cb3137cec5071d425540365 Mon Sep 17 00:00:00 2001 From: Leonid Kuligin Date: Wed, 5 Jun 2024 19:38:50 +0200 Subject: [PATCH] marked tests as extended (#281) --- libs/vertexai/tests/integration_tests/test_evaluation.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/libs/vertexai/tests/integration_tests/test_evaluation.py b/libs/vertexai/tests/integration_tests/test_evaluation.py index 9e680578..e5757e34 100644 --- a/libs/vertexai/tests/integration_tests/test_evaluation.py +++ b/libs/vertexai/tests/integration_tests/test_evaluation.py @@ -8,7 +8,7 @@ ) -@pytest.mark.release +@pytest.mark.extended def test_evaluate() -> None: evaluator = VertexStringEvaluator( metric="bleu", project_id=os.environ["PROJECT_ID"] @@ -28,7 +28,7 @@ def test_evaluate() -> None: assert result[1]["score"] < 1.0 -@pytest.mark.release +@pytest.mark.extended def test_evaluate_strings() -> None: evaluator = VertexStringEvaluator( metric="safety", project_id=os.environ["PROJECT_ID"] @@ -39,7 +39,7 @@ def test_evaluate_strings() -> None: assert "explanation" in result -@pytest.mark.release +@pytest.mark.extended async def test_aevaluate_strings() -> None: evaluator = VertexStringEvaluator( metric="question_answering_quality", project_id=os.environ["PROJECT_ID"] @@ -54,7 +54,7 @@ async def test_aevaluate_strings() -> None: assert "explanation" in result -@pytest.mark.release +@pytest.mark.extended async def test_evaluate_pairwise() -> None: evaluator = VertexPairWiseStringEvaluator( metric="pairwise_question_answering_quality",