1 files changed, 10 insertions, 5 deletions
diff --git a/app/services/ml/find_or_create_model_version_service.rb b/app/services/ml/find_or_create_model_version_service.rb
index f4d3f3e72d3..a5e9bf997cc 100644
--- a/app/services/ml/find_or_create_model_version_service.rb
+++ b/app/services/ml/find_or_create_model_version_service.rb
@@ -7,15 +7,20 @@ module Ml
       @name = params[:model_name]
       @version = params[:version]
       @package = params[:package]
+      @description = params[:description]
     end
 
     def execute
-      model = Ml::FindOrCreateModelService.new(project, name).execute
-      Ml::ModelVersion.find_or_create!(model, version, package)
-    end
+      model = Ml::FindOrCreateModelService.new(@project, @name).execute
+
+      model_version = Ml::ModelVersion.find_or_create!(model, @version, @package, @description)
 
-    private
+      model_version.candidate = ::Ml::CreateCandidateService.new(
+        model.default_experiment,
+        { model_version: model_version }
+      ).execute
 
-    attr_reader :version, :name, :project, :package
+      model_version
+    end
   end
 end