From 0c5e64825b3809b811d46d33d04f38f8e24711f2 Mon Sep 17 00:00:00 2001 From: Edward Park Date: Fri, 22 Sep 2023 08:30:36 -0700 Subject: [PATCH] add default Nones to optional props --- prefect_gcp/workers/vertex.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/prefect_gcp/workers/vertex.py b/prefect_gcp/workers/vertex.py index b000540..cbb12e0 100644 --- a/prefect_gcp/workers/vertex.py +++ b/prefect_gcp/workers/vertex.py @@ -114,6 +114,7 @@ class VertexAIWorkerVariables(BaseVariables): "See https://cloud.google.com/vertex-ai/docs/reference/rest/v1/MachineSpec" ), example="NVIDIA_TESLA_K80", + default=None, ) accelerator_count: Optional[int] = Field( title="Accelerator Count", @@ -122,6 +123,7 @@ class VertexAIWorkerVariables(BaseVariables): "See https://cloud.google.com/vertex-ai/docs/reference/rest/v1/MachineSpec" ), example=1, + default=None, ) boot_disk_type: str = Field( title="Boot Disk Type",