Registry / ai-ml / traceml

traceml

JSON →
library1.3.0pypypi✓ verified 81d ago

TraceML is the client-side engine for ML/Data tracking, visualization, dashboards, and model UI, designed specifically for integration with the Polyaxon MLOps platform. It allows users to log metrics, parameters, artifacts, and manage experiment runs programmatically. The current version is 1.3.0, and it maintains a relatively frequent release cadence, often aligning with Polyaxon platform updates.

pip install traceml
INSTALL
IMPORT
SIG · TRACEML
T
traceml
ai-mlpythonv1.3.0
Install
1.6s avg
Import
Disk
17MB
Pass rate
10/ 10
Env Coverage10 / 10
glibc
3.93.13
musl
3.93.13
Install & Compatibility
Where this runs
tested against v1.3.0 · pip install
no network on importno background threads
Install × environment matrix
Each cell = how many times install + import succeeded across repeated harness runs. Partial = flaky.
glibc = Debian/Ubuntu slim · musl = Alpine Linux
musl
py 3.103.910 runs
installs and imports cleanly · install 0.0s · import 0.000s · 19.1MB
glibc
py 3.103.910 runs
installs and imports cleanly · install 1.6s · import 0.000s · 20MB
17MB installed
● package 17MB
Code
Verified usage

Verified import paths — ran on the pinned version, not inferred.

run
import traceml
from traceml import run

Demonstrates how to initialize a TraceML run, log a metric, a parameter, and declare an artifact kind. By default, it runs in offline mode for local execution without a Polyaxon server. To track to a remote Polyaxon instance, ensure authentication is configured (e.g., via `POLYAXON_TOKEN` environment variable) and set `is_offline=False`.

import os from traceml.tracking import run from traceml.artifacts import V1ArtifactKind # Ensure Polyaxon is configured. For remote tracking, POLYAXON_TOKEN env var must be set # or ~/.polyaxon/config.yaml configured. For this quickstart to run standalone # without a Polyaxon server, 'is_offline=True' is used. # Log a simple metric and declare a model artifact with run.start( project="quickstart-traceml", name="my-first-experiment", tags=["python", "demo"], is_offline=True # Set to False to track to a Polyaxon server ) as r: print(f"Started run with UUID: {r.uuid}") r.log_metric("accuracy", 0.925) r.log_param("learning_rate", 0.001) # Log an artifact type. The actual model file would be uploaded later. r.log_outputs(model=V1ArtifactKind.MODEL) print("Logged metric, param, and model artifact kind.") print("Run finished.")
Debug
Known issues
gotchaTracking data not appearing in the Polyaxon UI or being incomplete.
fix
Ensure the `run.start()` context manager is used for automatic run finalization and artifact uploading. Verify Polyaxon server is running and accessible if tracking remotely. If `is_offline=True` is used, data is stored locally and will not appear in the Polyaxon UI. Check `polyaxon config show` for current client configuration.
affects: All versions
gotchaConfusion between `traceml.tracking.run` and `traceml.client.PolyaxonClient` for experiment tracking.
fix
`traceml.tracking.run` provides a higher-level, context-managed interface ideal for logging metrics, parameters, and artifacts within an experiment. Use `PolyaxonClient` for lower-level API interactions like managing projects, models, or users directly. For most experiment tracking tasks, `run` is the preferred entry point.
affects: All versions
gotchaIssues with authentication or connecting to the Polyaxon backend.
fix
Ensure the `POLYAXON_TOKEN` environment variable is set with a valid API token or `~/.polyaxon/config.yaml` is correctly configured with valid credentials and a reachable host. Verify network connectivity to the Polyaxon API endpoint.
affects: All versions
Upgrade
Version history
1.3.0latest on PyPI · released Apr 13, 2026
Audit
Dependencies

No dependency data recorded yet.

Agent activity
29 hits · last 30 days
node
26
OpenAI (training)
1
Resources
traceml — pip install traceml · libregistry