stormlog.mlflow_integration

Optional MLflow export helpers for Stormlog outputs (mirror of the W&B exporter).

class stormlog.mlflow_integration.MlflowExportConfig(enabled=False, tracking_uri=None, experiment=None, run_id=None, run_name=None, group=None, job_type=None, log_tables=True, log_artifacts=False, log_attribution=False)[source]

Bases: object

Runtime configuration for optional MLflow exports.

Parameters:

  • enabled (bool)

  • tracking_uri (str | None)

  • experiment (str | None)

  • run_id (str | None)

  • run_name (str | None)

  • group (str | None)

  • job_type (str | None)

  • log_tables (bool)

  • log_artifacts (bool)

  • log_attribution (bool)

enabled: bool = False
tracking_uri: str | None = None
experiment: str | None = None
run_id: str | None = None
run_name: str | None = None
group: str | None = None
job_type: str | None = None
log_tables: bool = True
log_artifacts: bool = False
log_attribution: bool = False
stormlog.mlflow_integration.add_mlflow_arguments(parser)[source]

Attach shared optional MLflow flags to a CLI parser.

Parameters:

parser (Any)

Return type:

None

stormlog.mlflow_integration.ensure_mlflow_available(config)[source]

Fail fast when the MLflow feature is enabled without dependencies installed.

Parameters:

config (MlflowExportConfig)

Return type:

None

stormlog.mlflow_integration.export_diagnose_bundle_to_mlflow(config, *, command_name, artifact_dir)[source]

Export one diagnose bundle directory to MLflow.

Parameters:
Return type:

None

stormlog.mlflow_integration.export_tracking_run_to_mlflow(config, *, command_name, session_summary, stats, events, output_path=None, telemetry_sink_dir=None, oom_dump_path=None, attribution_bundle_dir=None)[source]

Export one completed tracking session to MLflow.

Parameters:

  • config (MlflowExportConfig)

  • command_name (str)

  • session_summary (SessionSummary | None)

  • stats (Mapping[str, Any])

  • events (Sequence[Any])

  • output_path (str | Path | None)

  • telemetry_sink_dir (str | Path | None)

  • oom_dump_path (str | Path | None)

  • attribution_bundle_dir (str | Path | None)

Return type:

None

stormlog.mlflow_integration.mlflow_config_from_namespace(args)[source]

Build an MLflow export config from CLI args or a similar namespace.

Parameters:

args (Any)

Return type:

MlflowExportConfig