lib/graphql/schema/argument.rb in graphql-1.8.5 vs lib/graphql/schema/argument.rb in graphql-1.8.6

@@ -40,13 +40,20 @@
         @as = as
         @keyword = as || Schema::Member::BuildType.underscore(@name).to_sym
         @prepare = prepare
 
         if definition_block
-          instance_eval(&definition_block)
+          if definition_block.arity == 1
+            instance_exec(self, &definition_block)
+          else
+            instance_eval(&definition_block)
+          end
         end
       end
 
+      attr_writer :description
+
+      # @return [String] Documentation for this argument
       def description(text = nil)
         if text
           @description = text
         else
           @description