Fix integration tests (#552)

langchain-ai · Oct 11, 2024 · 9d997c5 · 9d997c5
1 parent 59380fb
commit 9d997c5
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/libs/vertexai/tests/integration_tests/test_callbacks.py b/libs/vertexai/tests/integration_tests/test_callbacks.py
@@ -10,7 +10,7 @@
 @pytest.mark.release
 @pytest.mark.parametrize(
     "model_name",
-    [_DEFAULT_MODEL_NAME, "code-bison@001"],
+    [_DEFAULT_MODEL_NAME],
 )
 def test_llm_invoke(model_name: str) -> None:
     vb = VertexAICallbackHandler()
@@ -30,7 +30,7 @@ def test_llm_invoke(model_name: str) -> None:
 @pytest.mark.release
 @pytest.mark.parametrize(
     "model_name",
-    [_DEFAULT_MODEL_NAME, "chat-bison@001", "codechat-bison@001"],
+    [_DEFAULT_MODEL_NAME],
 )
 def test_chat_call(model_name: str) -> None:
     vb = VertexAICallbackHandler()
@@ -51,7 +51,7 @@ def test_chat_call(model_name: str) -> None:
 @pytest.mark.release
 @pytest.mark.parametrize(
     "model_name",
-    [_DEFAULT_MODEL_NAME, "code-bison@001"],
+    [_DEFAULT_MODEL_NAME],
 )
 def test_invoke_config(model_name: str) -> None:
     vb = VertexAICallbackHandler()