debian-mirror-gitlab/app/controllers/projects/ml/experiments_controller.rb
2023-04-23 21:23:45 +05:30

49 lines
1.5 KiB
Ruby

# frozen_string_literal: true
module Projects
module Ml
class ExperimentsController < ::Projects::ApplicationController
include Projects::Ml::ExperimentsHelper
before_action :check_feature_flag
feature_category :mlops
MAX_EXPERIMENTS_PER_PAGE = 20
MAX_CANDIDATES_PER_PAGE = 30
def index
paginator = ::Ml::Experiment.by_project_id(@project.id)
.with_candidate_count
.keyset_paginate(cursor: params[:cursor], per_page: MAX_EXPERIMENTS_PER_PAGE)
@experiments = paginator.records
@page_info = page_info(paginator)
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?
find_params = params
.transform_keys(&:underscore)
.permit(:name, :order_by, :sort, :order_by_type)
paginator = CandidateFinder
.new(@experiment, find_params)
.execute
.keyset_paginate(cursor: params[:cursor], per_page: MAX_CANDIDATES_PER_PAGE)
@candidates = paginator.records.each(&:artifact_lazy)
@page_info = page_info(paginator)
end
private
def check_feature_flag
render_404 unless Feature.enabled?(:ml_experiment_tracking, @project)
end
end
end
end