fmt

langchain-ai · Mar 21, 2024 · 485cd3a · 485cd3a
1 parent 9c333a6
commit 485cd3a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/libs/vertexai/tests/unit_tests/test_llm.py b/libs/vertexai/tests/unit_tests/test_llm.py
@@ -10,10 +10,10 @@ def test_model_name() -> None:
     assert llm.model_name == "text-bison"
 
     for llm in [
-        VertexAI(model_name="foo"),
-        VertexAI(model="foo"),  # type: ignore[call-arg]
+        VertexAI(model_name="text-bison@001"),
+        VertexAI(model="text-bison@001"),  # type: ignore[call-arg]
     ]:
-        assert llm.model_name == "foo"
+        assert llm.model_name == "text-bison@001"
 
 
 def test_vertexai_args_passed() -> None: