blob: 891674adc2af5e452dd280a2c99d667a92a2712a (
plain)
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
|
# 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)
::Ml::Experiment.create!(name: name,
user: user,
project: project)
end
end
end
end
|
