From fd31bd1fc7954f69025d8e6bbe7f772ea9fb4bb0 Mon Sep 17 00:00:00 2001
From: GitLab Bot <gitlab-bot@gitlab.com>
Date: Tue, 11 Oct 2022 12:09:17 +0000
Subject: Add latest changes from gitlab-org/gitlab@master

---
 .../experiment_tracking/experiment_repository.rb   | 30 ++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 app/services/ml/experiment_tracking/experiment_repository.rb

(limited to 'app/services/ml/experiment_tracking/experiment_repository.rb')

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
-- 
cgit v1.2.3