reference_task should inherit from PythonTask

[pingsutw]

Tracking issue

https://flyte-org.slack.com/archives/CP2HDHKE1/p1722549808159589

Why are the changes needed?

AttributeError: 'ReferenceTask' object has no attribute '_container_image'

What changes were proposed in this pull request?

reference_task should just inherit from PythonTask since it could be a reference bigquery / non python function task.

How was this patch tested?

Unit test / pyflyte run

Setup process

Screenshots

import typing
from flytekit import task, workflow, reference_task, StructuredDataset


outputs = typing.NamedTuple("Outputs", results=StructuredDataset)


@reference_task(
    project="flyte-conformance",
    domain="development",
    name="bigquery",
    version="e2ZYwgh3ZzI7Q56rJjSN1g",
)
def ref_bq(version: int) -> outputs:
    ...


@reference_task(
    project="flyte-conformance",
    domain="development",
    name="flytesnacks.examples.basics.basics.workflow.slope",
    version="e2ZYwgh3ZzI7Q56rJjSN1g",
)
def ref_basic(x: typing.List[int], y: typing.List[int]) -> float:
    ...


@task(enable_deck=True)
def t1(a: int) -> int:
    return a + 1


@workflow
def wf() -> StructuredDataset:
    ref_basic(x=[1, 2, 3], y=[4, 5, 6])
    return ref_bq(version=1)

Check all the applicable boxes

• I updated the documentation accordingly.
• All new and existing tests passed.
• All commits are signed-off.

Related PRs

NA

Docs link

NA