Fixes failing test related to TaskNodeOverrides (#2230)

Signed-off-by: Thomas J. Fan <[email protected]>
flyteorg · Mar 1, 2024 · 784c08b · 784c08b
1 parent eb20459
commit 784c08b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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):