Simplify Virtus.included (that case statement made no sense)

kristianmandrup · Jun 8, 2012 · 15ad0b5 · 15ad0b5
1 parent 3615ab4
commit 15ad0b5
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/lib/virtus.rb b/lib/virtus.rb
@@ -25,11 +25,10 @@ module Virtus
   # @api private
   def self.included(object)
     super
-    case object
+    if Class === object
-    when Class  then object.send(:include, ClassInclusions)
+      object.send(:include, ClassInclusions)
-    when Module then object.extend(ModuleExtensions)
     else
-      raise ArgumentError, "Object not supported #{object.inspect}"
+      object.extend(ModuleExtensions)
     end
   end
   private_class_method :included