This is the Flyte SDK (v2) version of this plugin. The Flytekit version is available as flytekitplugins-mlflow.
MLflow tracking
flyteplugins-mlflow
MLflow tracking plugin for Flyte
pip install flyteplugins-mlflowQuick Start(example, may need adjustment)
pip install flyteplugins-mlflow
from flytekit import task, workflow
from flyteplugins.mlflow import mlflow_run, get_mlflow_run, mlflow_config, Mlflow
@task
def my_task() -> None:
mlflow_run(...)
@workflow
def my_workflow() -> None:
my_task()Available Imports (5)
Task for MLflow tracking.
from flyteplugins.mlflow import mlflow_run
Get the current MLflow run if within a `@mlflow_run` decorated task or trace.
from flyteplugins.mlflow import get_mlflow_run
Configuration type for MLflow tracking.
from flyteplugins.mlflow import mlflow_config
MLflow UI link for Flyte tasks.
extends dataclass — configuration or data structure for plugin setup
from flyteplugins.mlflow import Mlflow
Retrieve current MLflow configuration from Flyte context.
from flyteplugins.mlflow import get_mlflow_context
Dependencies
Related Plugins
Anthropic Claude
This plugin provides integration between Flyte and Anthropic's Claude API, enabling you to use Flyte tasks as tools for Claude agents.
Deck
This plugin provides additional renderers to improve task visibility within Flytekit.
Envd
envd is a command-line tool that helps you create the container-based development environment for AI/ML.
Flyte Interactive
FlyteInteractive plugin provides users' favorite interface to develop and debug a flyte task interactively. We support vscode, jupyter (WIP), and neovim (WIP).
Package Info
Downloads
Stats unavailable
