class ActiveRecord::Associations::BelongsToPolymorphicAssociation

def klass
  type = owner[reflection.foreign_type]
  type.presence && owner.class.polymorphic_class_for(type)
end