Mlflow

Package: flyteplugins.mlflow

MLflow UI link for Flyte tasks.

Resolves the link URL from one of two sources (in priority order):

1. Explicit link — set at definition or override time::

   @env.task(links=[Mlflow(link="https://mlflow.example.com/...")])
   
   task.override(links=[Mlflow(link="https://...")])()
   

2. Context link — auto-generated from link_host (and optional link_template) set via mlflow_config(). Propagates to child tasks that share or nest under the parent’s run. Cleared when a task creates an independent run (run_mode="new"). For nested runs (run_mode="nested"), the parent link is kept and the link name is automatically set to “MLflow (parent)”.

Parameters

class Mlflow(
    name: str,
    link: str,
    _decorator_run_mode: str,
)

Methods

get_link()

def get_link(
    run_name: str,
    project: str,
    domain: str,
    context: dict[str, str],
    parent_action_name: str,
    action_name: str,
    pod_name: str,
    kwargs,
) -> str

Returns a task log link given the action. Link can have template variables that are replaced by the backend.

Returns: The generated link.