chore: Fix chat streaming system tests

FUTURE_COPYBARA_INTEGRATE_REVIEW=#2450 from googleapis:release-please--branches--main d38964a PiperOrigin-RevId: 558616069
googleapis · Aug 21, 2023 · d6e076d · d6e076d
1 parent 73439c6
commit d6e076d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/system/aiplatform/test_language_models.py b/tests/system/aiplatform/test_language_models.py
@@ -306,7 +306,7 @@ def test_code_generation_streaming(self):
     def test_code_chat_model_send_message_streaming(self):
         aiplatform.init(project=e2e_base._PROJECT, location=e2e_base._LOCATION)
 
-        chat_model = language_models.ChatModel.from_pretrained("codeodechat-bison@001")
+        chat_model = language_models.ChatModel.from_pretrained("codechat-bison@001")
         chat = chat_model.start_chat()
 
         message1 = "Please help write a function to calculate the max of two numbers"