Fix test case that called gcp_corp_access.

PiperOrigin-RevId: 702033307
google · Dec 2, 2024 · 0666c0f · 0666c0f
1 parent db6f034
commit 0666c0f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/langfun/core/llms/vertexai_test.py b/langfun/core/llms/vertexai_test.py
@@ -426,6 +426,7 @@ def test_model_hub(self):
           model,
       )
 
+  @mock.patch.object(vertexai.VertexAIRest, 'credentials', new=True)
   def test_project_and_location_check(self):
     with self.assertRaisesRegex(ValueError, 'Please specify `project`'):
       _ = vertexai.VertexAIGeminiPro1()._api_initialized
@@ -496,6 +497,7 @@ def test_generation_config(self):
           lf.LMSamplingOptions(),
       )
 
+  @mock.patch.object(vertexai.VertexAIRest, 'credentials', new=True)
   def test_call_model(self):
     with mock.patch('requests.Session.post') as mock_generate:
       mock_generate.side_effect = mock_requests_post