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

Flyte

Following is my flyte wf file, can anyone help me to understand where i am doing wrong

import typing
import pandas as pd
import numpy as np
#from databricks import sql
import os
from flytekit import task, workflow
import flyte_db_plugin as fdp

#@task
def generate_normal_df() -&gt; pd.DataFrame:
    with fdp.DatabricksTask("","").get_sql("select * from student") as con:
#        result=con.sql_task("select * from student")
        return con

    # print(result)

    #  result= pd.DataFrame(i,columns=[""])
#    return result


#    for row in result:
        #print(row)
 #       return row
@task
def compute_stats(df:pd.DataFrame) -&gt; pd.DataFrame:
    return df

@workflow
def wf():
    return compute_stats(df=generate_normal_df())