diff options
Diffstat (limited to 'lib/api/ml/mlflow/model_versions.rb')
-rw-r--r-- | lib/api/ml/mlflow/model_versions.rb | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/lib/api/ml/mlflow/model_versions.rb b/lib/api/ml/mlflow/model_versions.rb index 4b211cf540c..53ba4ff36f1 100644 --- a/lib/api/ml/mlflow/model_versions.rb +++ b/lib/api/ml/mlflow/model_versions.rb @@ -27,13 +27,16 @@ module API desc: 'Register model under this name This field is required.' optional :description, type: String, desc: 'Optional description for model version.' + optional :tags, type: Array, desc: 'Additional metadata for a model version.' end post 'create', urgency: :low do present ::Ml::CreateModelVersionService.new( model, { model_name: params[:name], - description: params[:description] + description: params[:description], + metadata: params[:tags], + version: custom_version } ).execute, with: Entities::Ml::Mlflow::ModelVersion, |