Skip to content
This repository has been archived by the owner on Apr 26, 2024. It is now read-only.

Commit

Permalink
Apply suggestions from code review
Browse files Browse the repository at this point in the history
Co-authored-by: nate nowack <nate@prefect.io>
  • Loading branch information
desertaxle and zzstoatzz authored Dec 11, 2023
1 parent c84d722 commit 7abb9c7
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions prefect_gcp/aiplatform.py
Original file line number Diff line number Diff line change
Expand Up @@ -260,10 +260,10 @@ def get_corresponding_worker_type(self) -> str:

async def generate_work_pool_base_job_template(self) -> dict:
"""
Generate a base job template for a cloud-run work pool with the same
Generate a base job template for a `Vertex AI` work pool with the same
configuration as this block.
Returns:
- dict: a base job template for a cloud-run work pool
- dict: a base job template for a `Vertex AI` work pool
"""
base_job_template = await get_default_base_job_template_for_infrastructure_type(
self.get_corresponding_worker_type(),
Expand Down Expand Up @@ -312,7 +312,7 @@ async def generate_work_pool_base_job_template(self) -> dict:
base_job_template["variables"]["properties"][key]["default"] = value
else:
self.logger.warning(
f"Variable {key!r} is not supported by Cloud Run work pools."
f"Variable {key!r} is not supported by `Vertex AI` work pools."
" Skipping."
)

Expand Down

0 comments on commit 7abb9c7

Please sign in to comment.