blob: 158c26499b03bc9a90f4aa984173076de0567a06 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
# frozen_string_literal: true
FactoryBot.define do
factory :ml_models, class: '::Ml::Model' do
sequence(:name) { |n| "model#{n}" }
project
default_experiment { association :ml_experiments, project_id: project.id, name: name }
trait :with_versions do
versions { Array.new(2) { association(:ml_model_versions, model: instance) } }
end
trait :with_latest_version_and_package do
transient do
version { association(:ml_model_versions, :with_package, model: instance) }
end
versions { [version] }
latest_version { version }
end
end
end
|