retrieving-mlflow-traces
Retrieves MLflow traces using CLI or Python API. Use when the user asks to get a trace by ID, find traces, filter traces by status/tags/metadata/execution time, query traces, or debug failed traces. Triggers on "get trace", "search traces", "find failed traces", "filter traces by", "traces slower than", "query MLflow traces".
Install
With your agent
Install OMGSkills skill "mlflow/skills:retrieving-mlflow-traces" into this agent's normal skills directory: call the OMGSkills MCP get_install_plan tool for your agent and run the script it returns (it installs only the pinned commit and verifies SKILL.md), and stop if MCP or the pin is unavailable. Do not overwrite or change existing skills without asking me.
Manual command
git clone https://github.com/mlflow/skills /tmp/skills && ln -s /tmp/skills/retrieving-mlflow-traces ~/.claude/skills/retrieving-mlflow-traces
From README
Retrieving MLflow Traces Bound Unity Catalog Searches by Time When searching traces stored in Unity Catalog, include trace.timestampms >= in the filter string, with an upper bound when known. This also applies when searching by experiment ID, run ID, session, or other filters: a small result limit does not prevent a scan across the full trace history. In Python, pass the time condition through filterstring; mlflow.searchtraces() has no starttime argument. Use the time window requested by the user or established by the run or incident. For verification, record the time immediately before the instrumented run. Do not silently restrict historical investigations to the last hour. If the relevant period cannot be determined, ask for it before starting a broad UC search. The examples below demonstrate individual filters; combine them with the relevant time bounds for UC.
More from this repo
agent-evaluation
Use this when you need to EVALUATE OR IMPROVE or OPTIMIZE an existing LLM agent's output quality - including improving...
analyzing-mlflow-session
Analyzes an MLflow session — a sequence of traces from a multi-turn chat conversation or interaction.
analyzing-mlflow-trace
Analyzes a single MLflow trace to answer a user query about it.
