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

---
 .../ml/experiment_tracking/candidate_repository.rb | 85 ++++++++++++++++++++++
 .../experiment_tracking/experiment_repository.rb   | 30 ++++++++
 2 files changed, 115 insertions(+)
 create mode 100644 app/services/ml/experiment_tracking/candidate_repository.rb
 create mode 100644 app/services/ml/experiment_tracking/experiment_repository.rb

(limited to 'app/services/ml/experiment_tracking')

diff --git a/app/services/ml/experiment_tracking/candidate_repository.rb b/app/services/ml/experiment_tracking/candidate_repository.rb
new file mode 100644
index 00000000000..b6f87995185
--- /dev/null
+++ b/app/services/ml/experiment_tracking/candidate_repository.rb
@@ -0,0 +1,85 @@
+# frozen_string_literal: true
+
+module Ml
+  module ExperimentTracking
+    class CandidateRepository
+      attr_accessor :project, :user, :experiment, :candidate
+
+      def initialize(project, user)
+        @project = project
+        @user = user
+      end
+
+      def by_iid(iid)
+        ::Ml::Candidate.with_project_id_and_iid(project.id, iid)
+      end
+
+      def create!(experiment, start_time)
+        experiment.candidates.create!(
+          user: user,
+          start_time: start_time || 0
+        )
+      end
+
+      def update(candidate, status, end_time)
+        candidate.status = status.downcase if status
+        candidate.end_time = end_time if end_time
+
+        candidate.save
+      end
+
+      def add_metric!(candidate, name, value, tracked_at, step)
+        candidate.metrics.create!(
+          name: name,
+          value: value,
+          tracked_at: tracked_at,
+          step: step
+        )
+      end
+
+      def add_param!(candidate, name, value)
+        candidate.params.create!(name: name, value: value)
+      end
+
+      def add_metrics(candidate, metric_definitions)
+        return unless candidate.present?
+
+        metrics = metric_definitions.map do |metric|
+          {
+            candidate_id: candidate.id,
+            name: metric[:key],
+            value: metric[:value],
+            tracked_at: metric[:timestamp],
+            step: metric[:step],
+            **timestamps
+          }
+        end
+
+        ::Ml::CandidateMetric.insert_all(metrics, returning: false) unless metrics.empty?
+      end
+
+      def add_params(candidate, param_definitions)
+        return unless candidate.present?
+
+        parameters = param_definitions.map do |p|
+          {
+            candidate_id: candidate.id,
+            name: p[:key],
+            value: p[:value],
+            **timestamps
+          }
+        end
+
+        ::Ml::CandidateParam.insert_all(parameters, returning: false) unless parameters.empty?
+      end
+
+      private
+
+      def timestamps
+        current_time = Time.zone.now
+
+        { created_at: current_time, updated_at: current_time }
+      end
+    end
+  end
+end
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