Skip to content

Fixes MLflow run search compatibility

Eduardo Bonet requested to merge model_experiments/fix-search-endpoint into master

What does this MR do and why?

Fixes MLflow run search compatibility

Changes the http method of runs/search to POST instead of GET, as defined in the mlflow rest api https://mlflow.org/docs/1.28.0/rest-api.html#search-runs

Changelog: fixed

How to set up and validate locally

  1. Create an experiment and a few candidates:

    project_id = 1
user_id = 1
exp = Ml::Experiment.create!(name: "Gitlab Experiment", user_id: user_id, project_id: project_id)
exp.candidates.create!(user_id: user_id, start_time: 0, internal_id: 1, project_id: project_id)
exp.candidates.create!(user_id: user_id, start_time: 0, internal_id: 1, project_id: project_id)

  2. Create a request to fetch the the candidates/runs (set experiment_ids[]=? is probably 1, but if you created other experiments before this could change):

    curl -X POST -H "Authorization: Bearer $GPAT" "http://localhost:3000/api/v4/projects/$PROJECT_ID/ml/mlflow/api/2.0/mlflow/runs/search?experiment_ids[]=1"

MR acceptance checklist

This checklist encourages us to confirm any changes have been analyzed to reduce risks in quality, performance, reliability, security, and maintainability.

Merge request reports