Flyte enables production-grade orchestration for machine learning workflows and data processing created to accelerate local workflows to production.

Flyte

Hello Flyte community. I have a question about shell tasks. I would like to set the container_image in shell task dynamically

would this work or is there a better way? up until now I have used a string for container_image and that worked well. now depending on the workflow I would like to specify a different image
```test_task = ShellTask(
    name="test",
    debug=True,
    container_image=kwtypes(image_uri=str),
    script="""
    {inputs.command} &gt;&gt; {outputs.stdout} 2&gt;&gt; {outputs.stderr}
    """,
    inputs=kwtypes(command=str),```

<https://docs.flyte.org/en/latest/user_guide/productionizing/customizing_task_resources.html#resource-with-overrides>

just following up on this. <@U073L5Y9GJD>, were you able to give this a try?


yes it seems to work with a quick test thank you