Add latest changes from gitlab-org/gitlab@14-10-stable-eev14.10.0-rc42

1 files changed, 6 insertions, 3 deletions

diff --git a/lib/bulk_imports/stage.rb b/lib/bulk_imports/stage.rb
index 9c19e9ea60b..6cf394c5df0 100644
--- a/lib/bulk_imports/stage.rb
+++ b/lib/bulk_imports/stage.rb
@@ -2,10 +2,13 @@
 
 module BulkImports
   class Stage
-    def initialize(bulk_import)
-      raise(ArgumentError, 'Expected an argument of type ::BulkImport') unless bulk_import.is_a?(::BulkImport)
+    def initialize(bulk_import_entity)
+      unless bulk_import_entity.is_a?(::BulkImports::Entity)
+        raise(ArgumentError, 'Expected an argument of type ::BulkImports::Entity')
+      end
 
-      @bulk_import = bulk_import
+      @bulk_import_entity = bulk_import_entity
+      @bulk_import = bulk_import_entity.bulk_import
     end
 
     def pipelines