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

# frozen_string_literal: true

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

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

      Ml::ModelVersion.find_or_create!(model, version, package)
    end

    private

    attr_reader :version, :name, :project, :package
  end
end