From 784c08b36f5280036b3f680fcff894890d2bca0e Mon Sep 17 00:00:00 2001 From: "Thomas J. Fan" Date: Fri, 1 Mar 2024 16:12:06 -0500 Subject: [PATCH] Fixes failing test related to TaskNodeOverrides (#2230) Signed-off-by: Thomas J. Fan --- flytekit/models/core/workflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/flytekit/models/core/workflow.py b/flytekit/models/core/workflow.py index aef5d3c46f..44fe7e1f44 100644 --- a/flytekit/models/core/workflow.py +++ b/flytekit/models/core/workflow.py @@ -598,7 +598,7 @@ def __init__( self, resources: typing.Optional[Resources], extended_resources: typing.Optional[tasks_pb2.ExtendedResources], - container_image: typing.Optional[str], + container_image: typing.Optional[str] = None, ): self._resources = resources self._extended_resources = extended_resources @@ -613,7 +613,7 @@ def extended_resources(self) -> tasks_pb2.ExtendedResources: return self._extended_resources @property - def container_image(self) -> str: + def container_image(self) -> typing.Optional[str]: return self._container_image def to_flyte_idl(self):