Skip to content

Commit 0663d8e

Browse files
Add escape hatch for reference entities in remote._get_image_names (#3341)
Signed-off-by: Yee Hing Tong <[email protected]>
1 parent a693622 commit 0663d8e

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

flytekit/remote/remote.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1330,7 +1330,11 @@ def _version_from_hash(
13301330
return base64.urlsafe_b64encode(h.digest()).decode("ascii").rstrip("=")
13311331

13321332
@staticmethod
1333-
def _get_image_names(entity: typing.Union[PythonAutoContainerTask, WorkflowBase]) -> typing.List[str]:
1333+
def _get_image_names(
1334+
entity: typing.Union[PythonAutoContainerTask, WorkflowBase, LaunchPlan, ReferenceEntity],
1335+
) -> typing.List[str]:
1336+
if isinstance(entity, ReferenceEntity):
1337+
return []
13341338
if isinstance(entity, PythonAutoContainerTask) and isinstance(entity.container_image, ImageSpec):
13351339
return [entity.container_image.image_name()]
13361340
if isinstance(entity, WorkflowBase):

0 commit comments

Comments
 (0)