mysterious-painter-66441
09/18/2024, 4:59 PMfrom 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