Add latest changes from gitlab-org/gitlab@master

1 files changed, 30 insertions, 0 deletions

diff --git a/app/services/ml/experiment_tracking/experiment_repository.rb b/app/services/ml/experiment_tracking/experiment_repository.rb
new file mode 100644
index 00000000000..891674adc2a
--- /dev/null
+++ b/app/services/ml/experiment_tracking/experiment_repository.rb
@@ -0,0 +1,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