Skip to content

Commit

Permalink
fix task invocation default param
Browse files Browse the repository at this point in the history
  • Loading branch information
SebastianMorawiec committed Oct 8, 2024
1 parent 69648f2 commit bccb2fc
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -670,6 +670,7 @@ def __call__(
custom_name=parse_custom_name(self._custom_name, signature),
custom_image=self._custom_image,
env_vars=self._env_vars,
type="task_invocation",
)
)

Expand Down Expand Up @@ -735,7 +736,7 @@ def __init__(
custom_name: Optional[str],
custom_image: Optional[str],
env_vars: Optional[Dict[str, str]],
type: str = "task_invocation",
type: str,
):
self.task = task
self.args = args
Expand Down Expand Up @@ -939,6 +940,7 @@ def with_invocation_meta(
custom_name=invocation.custom_name,
custom_image=new_custom_image,
env_vars=new_env_vars,
type=invocation.type,
)

return ArtifactFuture(
Expand Down

0 comments on commit bccb2fc

Please sign in to comment.