Skip to content

Mlflow Model Version `update` Endpoint

Sri Rang requested to merge mlflow-api-model-versions-endpoint-update into master

What does this MR do and why?

Implements the update endpoint for Mlflow::ModelVersions, thus users can update model_version description.

Screenshots or screen recordings

None, purely backend

Database

Update query

UPDATE "ml_model_versions"
SET "description" = 'my model description'
WHERE "ml_model_versions"."id" = 1

Local explain analyze

Update on ml_model_versions  (cost=0.15..2.17 rows=0 width=0) (actual time=0.092..0.093 rows=0 loops=1)
  ->  Index Scan using ml_model_versions_pkey on ml_model_versions  (cost=0.15..2.17 rows=1 width=38) (actual time=0.027..0.029 rows=1 loops=1)
        Index Cond: (id = 1)
Planning Time: 0.118 ms
Execution Time: 0.138 ms

How to set up and validate locally

Verify the tests that hit the API endpoint and get desired result compliant with Mlflow schema

MR acceptance checklist

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

Edited by Sri Rang

Merge request reports

Loading