File: create_candidate_service.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (37 lines) | stat: -rw-r--r-- 899 bytes parent folder | download
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
# frozen_string_literal: true

module Ml
  class CreateCandidateService
    def initialize(experiment, params = {})
      @experiment = experiment
      @name = params[:name]
      @user = params[:user]
      @start_time = params[:start_time]
      @model_version = params[:model_version]
    end

    def execute
      Ml::Candidate.create!(
        experiment: experiment,
        project: experiment.project,
        name: candidate_name,
        start_time: start_time || 0,
        user: user,
        model_version: model_version
      )
    end

    private

    def candidate_name
      name.presence || random_candidate_name
    end

    def random_candidate_name
      parts = Array.new(3).map { FFaker::AnimalUS.common_name.downcase.delete(' ') } << rand(10000)
      parts.join('-').truncate(255)
    end

    attr_reader :name, :user, :experiment, :start_time, :model_version
  end
end