lib/rocket_job/batch.rb in rocketjob-5.4.1 vs lib/rocket_job/batch.rb in rocketjob-6.0.0.rc1

@@ -5,19 +5,22 @@
 require "rocket_job/batch/model"
 require "rocket_job/batch/state_machine"
 require "rocket_job/batch/throttle"
 require "rocket_job/batch/throttle_running_workers"
 require "rocket_job/batch/worker"
+# Ensure after_perform is run first and #upload override is after IO#upload is defined.
+require "rocket_job/batch/categories"
 
 module RocketJob
   module Batch
     extend ActiveSupport::Concern
 
     include Model
     include StateMachine
     include Callbacks
     include Logger
     include Worker
+    include Categories
     include Throttle
     include ThrottleRunningWorkers
     include IO
 
     autoload :LowerPriority, "rocket_job/batch/lower_priority"