lib/arel/extensions/table.rb in arel_toolkit-0.4.6 vs lib/arel/extensions/table.rb in arel_toolkit-0.4.7

@@ -13,19 +13,24 @@
       attr_accessor :relpersistence
 
       def initialize(
         name,
         as: nil,
-        type_caster: nil,
+        klass: nil,
+        type_caster: klass&.type_caster,
         only: false,
         schema_name: nil,
         relpersistence: 'p'
       )
         @only = only
         @schema_name = schema_name
         @relpersistence = relpersistence
 
-        super(name, as: as, type_caster: type_caster)
+        if Gem.loaded_specs['activerecord'].version < Gem::Version.new('6.1.0')
+          super(name, as: as, type_caster: type_caster)
+        else
+          super(name, klass: klass, as: as, type_caster: type_caster)
+        end
       end
     end
 
     prepend TableExtension
   end