orange-army-92143
03/27/2025, 4:54 PMconfig.yml
? This is not clear at all in the docs (also not sure if it's even possible)average-finland-92144
03/27/2025, 5:08 PMconfig.yaml
as that only controls client -> control plane communication and the container image for executions is at the dataplane level.
You could use partial
to do something like this:
image = ImageSpec(
name="my-image",
packages=[
"scikit-learn",
],
)
task = partial(
flytekit.task,
container_image=image,
)
And then use the decorator as usual:
@task
def my_function() -> None
...
average-finland-92144
03/27/2025, 5:09 PMorange-army-92143
03/27/2025, 5:09 PM