"""
Reference Task
--------------

A :py:func:`flytekit.reference_task` references the Flyte tasks that have already been defined, serialized, and registered.
You can reference tasks from other projects and create a workflow that uses tasks declared by others.
These tasks can be in their own containers, python runtimes, flytekit versions, and even different languages.

This example demonstrates the usage of reference tasks.

.. note::
    Reference tasks cannot be run locally. You must mock them out.
"""

# %%
# First, let's import the dependencies.
from typing import List

from flytekit import reference_task, workflow
from flytekit.types.file import FlyteFile


# %%
# We define a ``reference_task`` by importing the ``normalize_columns()`` function.
@reference_task(
    project="flytesnacks",
    domain="development",
    name="core.flyte_basics.files.normalize_columns",
    version="{{ registration.version }}",
)
def normalize_columns(
    csv_url: FlyteFile,
    column_names: List[str],
    columns_to_normalize: List[str],
    output_location: str,
) -> FlyteFile:
    ...


# %%
# .. note::
#     The Macro ``{{ registration.version }}`` is filled during the registration time by `flytectl register`. This is usually not required for reference tasks; you should
#     ideally bind to a specific version of the entity -- task / launchplan. But in the case of this example, we are registering both the actual task ``core.flyte_basics.files.rotate`` and
#     the workflow that references it. Thus we want it to actually be updated to the version of a specific release of Flytesnacks. This is why we use the ``{{ registration.version }}`` macro.
#
#     A typical example of reference task would look more like this:
#
#     .. code-block:: python
#
#          @reference_task(
#               project="flytesnacks",
#               domain="development",
#               name="core.flyte_basics.files.normalize_columns",
#               version="d06cebcfbeabc02b545eefa13a01c6ca992940c8", # If using GIT for versioning OR 0.16.0, if semver
#           )
#           def normalize_columns(...):
#               ...

# %%
# Next, we define a workflow to call the ``rotate()`` task.
@workflow
def wf() -> FlyteFile:
    return normalize_columns(
        csv_url="https://people.sc.fsu.edu/~jburkardt/data/csv/biostats.csv",
        column_names=["Name", "Sex", "Age", "Heights (in)", "Weight (lbs)"],
        columns_to_normalize=["Age"],
        output_location="",
    )


# %%
# Lastly, we can trigger the workflow in a sandbox environment.