1 files changed, 16 insertions, 0 deletions
diff --git a/spec/factories/ml/model_versions.rb b/spec/factories/ml/model_versions.rb
new file mode 100644
index 00000000000..5ae0446b78d
--- /dev/null
+++ b/spec/factories/ml/model_versions.rb
@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+
+FactoryBot.define do
+  factory :ml_model_versions, class: '::Ml::ModelVersion' do
+    sequence(:version) { |n| "version#{n}" }
+
+    model { association :ml_models }
+    project { model.project }
+
+    trait :with_package do
+      package do
+        association :ml_model_package, name: model.name, version: version, project_id: project.id
+      end
+    end
+  end
+end