Skip to content

Draft: Model Versions API Dump

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

Implementing the MLFlow APIs for ModelVersion as requested here: #427454

This MR goes "breadth-first" i.e. to define all the endpoints, ensure the API schema is identical to MLFlow API and provide mock responses.

ModelVersion endpoints introduced:

  • /create
  • /get
  • /delete
  • /update Missing from original issue, added upon review
  • /get-download-uri
  • /search

Entities for mapped responses to MlFlow API:

  • Entities::MlFlow::ModelVersions::Responses::Create
  • Entities::MlFlow::ModelVersions::Responses::Get
  • Entities::MlFlow::ModelVersions::Responses::Delete
  • Entities::MlFlow::ModelVersions::Responses::Update
  • Entities::MlFlow::ModelVersions::Responses::GetArtifactUri
  • Entities::MlFlow::ModelVersions::Responses::Search

Entities for mapping core MLFlow API types

  • Entities::MlFlow::ModelVersions::Types::ModelVersion
  • Entities::MlFlow::ModelVersions::Types::ModelVersionTag

Services with mock implementation

  • Ml::ModelVersions::CreateModelVersionService
  • Ml::ModelVersions::GetModelVersionService
  • Ml::ModelVersions::DeleteModelVersionService
  • Ml::ModelVersions::UpdateModelVersionService
  • Ml::ModelVersions::GetArtifactUriService
  • Ml::ModelVersions::SearchModelVersionsService

These services return mock responses that are mapped to MlFlow response entities.

In-progress

  • Tests
Edited by Sri Rang

Merge request reports