# pip install flytekitplugins-polars

from flytekit import task, workflow
import polars as pl
import pandas as pd

@task
def t1() -> pl.DataFrame:
    return pl.DataFrame(pd.DataFrame({"a": [1, 2, 3]}))

@task
def t2(df: pl.DataFrame) -> int:
    return len(df)

@workflow
def wf() -> int:
    df = t1()
    return t2(df=df)

wf()