app/services/ml/find_or_create_model_version_service.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

# frozen_string_literal: true

module Ml
  class FindOrCreateModelVersionService
    def initialize(project, params = {})
      @project = project
      @name = params[:model_name]
      @version = params[:version]
      @package = params[:package]
      @description = params[:description]
    end

    def execute
      model = Ml::FindOrCreateModelService.new(@project, @name).execute

      model_version = Ml::ModelVersion.find_or_create!(model, @version, @package, @description)

      model_version.candidate = ::Ml::CreateCandidateService.new(
        model.default_experiment,
        { model_version: model_version }
      ).execute

      model_version
    end
  end
end