Frank Shen
04/28/2023, 10:25 PMEvan Sadler
04/28/2023, 11:02 PMflytekitplugins-polars
? It has a type transformer for polars dataframes!# 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()
Frank Shen
05/01/2023, 11:49 PMEvan Sadler
05/02/2023, 4:57 PM