reference_task_target.py

"""Target workflow for the reference_task / reference_launch_plan examples.

Run this first — it deploys ``greet_env.greet`` and ``flytekit_workflow.greet_wf``
to the cluster, which the reference examples then invoke. Note: tasks must be
*deployed* (not just run) to be resolvable by reference.
"""

import flyte_migrate  # noqa: F401, I001

from flytekit import task, workflow


@task
def greet(name: str) -> str:
    return f"Hello, {name}!"


@workflow
def greet_wf(name: str) -> str:
    return greet(name=name)


if __name__ == "__main__":
    from pathlib import Path

    import flyte

    # Pin the bundle root to the repo root (same as v2's deploy.py pattern) so the
    # code bundle includes src/flyte_migrate regardless of where this is run from.
    flyte.init_from_config(root_dir=Path(__file__).parent.parent)
    # Deploy the same way as v2: the workflow's parent env depends on every
    # shimmed task env, so one flyte.deploy call registers everything.
    deployments = flyte.deploy(greet_wf.parent_env())
    print(deployments[0].summary_repr())