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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
# frozen_string_literal: true
module Ml
class Model < ApplicationRecord
include Presentable
include Sortable
include CacheMarkdownField
EXPERIMENT_NAME_PREFIX = '[model]'
validates :project, :default_experiment, presence: true
validates :name,
format: Gitlab::Regex.ml_model_name_regex,
uniqueness: { scope: :project },
presence: true,
length: { maximum: 255 }
validate :valid_default_experiment?
validates :description,
length: { maximum: 10_000 }
has_one :default_experiment, class_name: 'Ml::Experiment'
belongs_to :project
belongs_to :user
has_many :versions, class_name: 'Ml::ModelVersion'
has_many :candidates, -> { without_model_version }, class_name: 'Ml::Candidate', through: :default_experiment
has_many :metadata, class_name: 'Ml::ModelMetadata'
has_one :latest_version, -> { latest_by_model }, class_name: 'Ml::ModelVersion', inverse_of: :model
scope :including_latest_version, -> { includes(:latest_version) }
scope :including_project, -> { includes(:project) }
scope :with_version_count, -> {
left_outer_joins(:versions)
.select("ml_models.*, count(ml_model_versions.id) as version_count")
.group(:id)
}
scope :by_name, ->(name) { where("ml_models.name LIKE ?", "%#{sanitize_sql_like(name)}%") } # rubocop:disable GitlabSecurity/SqlInjection
scope :by_project, ->(project) { where(project_id: project.id) }
cache_markdown_field :description
def all_packages
Packages::MlModel::Package.where(project: project, id: versions.select(:package_id))
end
def valid_default_experiment?
return unless default_experiment
errors.add(:default_experiment) unless default_experiment.name == self.class.prefixed_experiment(name)
errors.add(:default_experiment) unless default_experiment.project_id == project_id
end
def self.by_project_id_and_id(project_id, id)
find_by(project_id: project_id, id: id)
end
def self.by_project_id_and_name(project_id, name)
find_by(project_id: project_id, name: name)
end
def self.prefixed_experiment(string)
"#{EXPERIMENT_NAME_PREFIX}#{string}"
end
end
end
|
