Hi, I'm reading the Flyte documentation on extending a task. There are two ways listed.

@task(task_config=MyContainerExecutionTask(
    plugin_specific_config_a=...,
    plugin_specific_config_b=...,
    ...
))
def foo(...) -> ...:
    ...

And

query_task = SnowflakeTask(
    query="Select * from x where x.time < {{.inputs.time}}",
    inputs=kwtypes(time=datetime),
    output_schema_type=pandas.DataFrame,
)

@workflow
def my_wf(t: datetime) -> ...:
    df = query_task(time=t)
    return process(df=df)

I'm trying to find examples/tutorials/how-tos on the first approach, but I can't find any. Can someone point me to an example?

Does this work https://docs.flyte.org/projects/cookbook/en/stable/auto/core/extend_flyte/user_container.html

Thanks! I also found the AWS Batch example, which is closer to what I need

Fantastic

Yeah sorry, I realized my question isn't exactly what I was looking for. Still learning flyte 😅

No worries 😍

Welcome Abdullah, we use backend plugins for some other tasks, it might work for you as well. can sync