pass in creds as config variable

PrefectHQ · Sep 19, 2023 · 583210f · 583210f
1 parent 54fa575
commit 583210f
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/test_vertex_worker.py b/tests/test_vertex_worker.py
@@ -54,11 +54,10 @@ async def test_validate_empty_job_spec(self, gcp_credentials):
         base_job_template = VertexAIWorker.get_default_base_job_template()
         base_job_template["job_configuration"]["job_spec"] = {}
         base_job_template["job_configuration"]["region"] = "us-central1"
-        base_job_template["job_configuration"]["credentials"] = gcp_credentials
 
         with pytest.raises(pydantic.ValidationError) as excinfo:
             await VertexAIWorkerJobConfiguration.from_template_and_values(
-                base_job_template, {}
+                base_job_template, {"credentials": gcp_credentials}
             )
 
         assert excinfo.value.errors() == [
@@ -87,11 +86,10 @@ async def test_validate_incomplete_worker_pool_spec(self, gcp_credentials):
             ],
         }
         base_job_template["job_configuration"]["region"] = "us-central1"
-        base_job_template["job_configuration"]["credentials"] = gcp_credentials
 
         with pytest.raises(pydantic.ValidationError) as excinfo:
             await VertexAIWorkerJobConfiguration.from_template_and_values(
-                base_job_template, {}
+                base_job_template, {"credentials": gcp_credentials}
             )
 
         assert excinfo.value.errors() == [