summaryrefslogtreecommitdiff
path: root/app/controllers/projects/ml/experiments_controller.rb
blob: 1e1c4b1587cdc347fef7ef477cdc813efa3a23c8 (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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
# frozen_string_literal: true

module Projects
  module Ml
    class ExperimentsController < ::Projects::ApplicationController
      before_action :check_feature_flag

      feature_category :mlops

      MAX_EXPERIMENTS_PER_PAGE = 20
      MAX_CANDIDATES_PER_PAGE = 30

      def index
        @experiments = ::Ml::Experiment.by_project_id(@project.id).page(params[:page]).per(MAX_EXPERIMENTS_PER_PAGE)
      end

      def show
        @experiment = ::Ml::Experiment.by_project_id_and_iid(@project.id, params[:id])

        return redirect_to project_ml_experiments_path(@project) unless @experiment.present?

        page = params[:page].to_i
        page = 1 if page == 0

        @candidates = @experiment.candidates
                                 .including_relationships
                                 .page(page)
                                 .per(MAX_CANDIDATES_PER_PAGE)

        return unless @candidates

        return redirect_to(url_for(page: @candidates.total_pages)) if @candidates.out_of_range?

        @pagination = {
          page: page,
          is_last_page: @candidates.last_page?,
          per_page: MAX_CANDIDATES_PER_PAGE,
          total_items: @candidates.total_count
        }

        @candidates.each(&:artifact_lazy)
      end

      private

      def check_feature_flag
        render_404 unless Feature.enabled?(:ml_experiment_tracking, @project)
      end
    end
  end
end