lib/rocket_job/extensions/mongoid/factory.rb in rocketjob-5.4.1 vs lib/rocket_job/extensions/mongoid/factory.rb in rocketjob-6.0.0.rc1

@@ -1,21 +1,13 @@
 require "mongoid/factory"
 
 module RocketJob
   # Don't convert to Mongoid::Factory since it conflicts with Mongoid use.
   module MongoidFactory
-    if Mongoid::VERSION.to_f >= 7.1
-      def from_db(klass, attributes = nil, criteria = nil, selected_fields = nil)
-        obj                 = super(klass, attributes, criteria, selected_fields)
-        obj.collection_name = criteria.collection_name if criteria
-        obj
-      end
-    else
-      def from_db(klass, attributes = nil, criteria = nil)
-        obj                 = super(klass, attributes, criteria)
-        obj.collection_name = criteria.collection_name if criteria
-        obj
-      end
+    def from_db(klass, attributes = nil, criteria = nil, selected_fields = nil)
+      obj                 = super(klass, attributes, criteria, selected_fields)
+      obj.collection_name = criteria.collection_name if criteria
+      obj
     end
   end
 end
 
 ::Mongoid::Factory.extend(RocketJob::MongoidFactory)