Hi community! Is there a way to use task outputs t...
# flyte-support
m
Hi community! Is there a way to use task outputs to dynamically set the container image for subsequent tasks? I am trying to use the with_overrides, but it is not working.
Copy code
from flytekit import task, workflow
@task()
def get_image(pandas_version: str) -> str:
    return f"repo/pandas:{pandas_version}"


@task()
def check_pandas_version(pandas_version: str) -> str:
    import pandas as pd
    return pd.__version__


@workflow
def wf(pandas_version: str) -> str:
    image_str = get_image(pandas_version=pandas_version)
    installed_pandas_version=check_pandas_version(pandas_version=pandas_version).with_overrides(container_image=image_str)
    return installed_pandas_version