mantik.runs.mlflow_monkeypatch

MLflow local run function.

This source file contains code from [MLflow](https://github.com/mlflow/mlflow) licensed under Apache-2.0 license, see [here](https://github.com/mlflow/mlflow/blob/1eef4641df6f605b7d9faa83b0fc25e65877dbf4/LICENSE.txt) for the original license.

Changes made to the original source code are denoted as such with comments.

For Mantik developers NOTE: this has to be kept up to date with the different Mlflow versions.

If any changes occur in the original code that can be found here: https://github.com/mlflow/mlflow/blob/master/mlflow/projects/backend/local.py

Functions

run_entry_point(command, work_dir, experiment_id, run_id)

Run an entry point command in a subprocess,

run_mlflow_run_cmd(mlflow_run_arr, env_map)

Invoke mlflow run in a subprocess,

Module Contents

mantik.runs.mlflow_monkeypatch.run_entry_point(command, work_dir, experiment_id, run_id)[source]

Run an entry point command in a subprocess, returning a SubmittedRun that can be used to query the run’s status.

Parameters:
  • command – Entry point command to run

  • work_dir – Working directory in which to run the command

  • run_id – MLflow run ID associated with the entry point execution.

mantik.runs.mlflow_monkeypatch.run_mlflow_run_cmd(mlflow_run_arr, env_map)[source]

Invoke mlflow run in a subprocess, which in turn runs the entry point in a child process. Returns a handle to the subprocess. Popen launched to invoke mlflow run.