nit

Signed-off-by: Kevin Su <[email protected]>
flyteorg · May 4, 2023 · 25076fa · 25076fa
1 parent 33f8a3c
commit 25076fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flytekit/core/pod_template.py b/flytekit/core/pod_template.py
@@ -13,7 +13,7 @@
 class PodTemplate(object):
     """Custom PodTemplate specification for a Task."""
 
-    pod_spec: "V1PodSpec" = None
+    pod_spec: Optional["V1PodSpec"] = None
     primary_container_name: str = PRIMARY_CONTAINER_DEFAULT_NAME
     labels: Optional[Dict[str, str]] = None
     annotations: Optional[Dict[str, str]] = None