From efe06febf2447be2e3563b08086ad1763f32a416 Mon Sep 17 00:00:00 2001 From: Leonid Kuligin Date: Fri, 11 Oct 2024 20:21:10 +0200 Subject: [PATCH] remove codechat from integration tests --- libs/vertexai/tests/integration_tests/test_chat_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/vertexai/tests/integration_tests/test_chat_models.py b/libs/vertexai/tests/integration_tests/test_chat_models.py index fedcfb56..6e9c554b 100644 --- a/libs/vertexai/tests/integration_tests/test_chat_models.py +++ b/libs/vertexai/tests/integration_tests/test_chat_models.py @@ -42,7 +42,7 @@ from langchain_google_vertexai.chat_models import _parse_chat_history_gemini from tests.integration_tests.conftest import _DEFAULT_MODEL_NAME -model_names_to_test = ["codechat-bison", _DEFAULT_MODEL_NAME] +model_names_to_test = [_DEFAULT_MODEL_NAME] rate_limiter = InMemoryRateLimiter(requests_per_second=1.0)