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
|
# frozen_string_literal: true
module Ml
module ExperimentTracking
class ExperimentRepository
attr_accessor :project, :user
def initialize(project, user = nil)
@project = project
@user = user
end
def by_iid_or_name(iid: nil, name: nil)
return ::Ml::Experiment.by_project_id_and_iid(project.id, iid) if iid
::Ml::Experiment.by_project_id_and_name(project.id, name) if name
end
def all
::Ml::Experiment.by_project_id(project.id)
end
def create!(name, tags = nil)
experiment = ::Ml::Experiment.create!(name: name,
user: user,
project: project)
add_tags(experiment, tags)
experiment
end
def add_tag!(experiment, key, value)
return unless experiment.present?
experiment.metadata.create!(name: key, value: value)
end
private
def timestamps
current_time = Time.zone.now
{ created_at: current_time, updated_at: current_time }
end
def add_tags(experiment, tag_definitions)
return unless experiment.present? && tag_definitions.present?
entities = tag_definitions.map do |d|
{
experiment_id: experiment.id,
name: d[:key],
value: d[:value],
**timestamps
}
end
::Ml::ExperimentMetadata.insert_all(entities, returning: false) unless entities.empty?
end
end
end
end
|
