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
66
67
68
69
70
71
72
73
74
75
76
|
# frozen_string_literal: true
module Ml
class Experiment < ApplicationRecord
include AtomicInternalId
include Sortable
PACKAGE_PREFIX = 'ml_experiment_'
validates :name, :project, presence: true
validates :name, uniqueness: { scope: :project, message: "should be unique in the project" }
belongs_to :project
belongs_to :user
belongs_to :model, optional: true, inverse_of: :default_experiment
has_many :candidates, class_name: 'Ml::Candidate'
has_many :metadata, class_name: 'Ml::ExperimentMetadata'
scope :including_project, -> { includes(:project) }
scope :by_project, ->(project) { where(project: project) }
scope :with_candidate_count, -> {
left_outer_joins(:candidates)
.select("ml_experiments.*, count(ml_candidates.id) as candidate_count")
.group(:id)
}
has_internal_id :iid, scope: :project
before_destroy :stop_destroy
def package_name
"#{PACKAGE_PREFIX}#{iid}"
end
def stop_destroy
return unless model_id
errors[:base] << "Cannot delete an experiment associated to a model"
# According to docs, throw is the correct way to stop on a callback
# https://api.rubyonrails.org/classes/ActiveRecord/Callbacks.html#module-ActiveRecord::Callbacks-label-Canceling+callbacks
throw :abort # rubocop:disable Cop/BanCatchThrow
end
class << self
def by_project_id_and_iid(project_id, iid)
find_by(project_id: project_id, iid: iid)
end
def by_project_id_and_name(project_id, name)
find_by(project_id: project_id, name: name)
end
def by_project_id(project_id)
where(project_id: project_id).order(id: :desc)
end
def package_for_experiment?(package_name)
return false unless package_name&.starts_with?(PACKAGE_PREFIX)
iid = package_name.delete_prefix(PACKAGE_PREFIX)
numeric?(iid)
end
def find_or_create(project, name, user)
create_with(user: user).find_or_create_by(project: project, name: name)
end
private
def numeric?(value)
value.match?(/\A\d+\z/)
end
end
end
end
|