lib/riif/iif.rb in riif-0.5.0 vs lib/riif/iif.rb in riif-0.6.0

@@ -24,10 +24,14 @@
 module Riif
   class IIF
     def initialize(&block)
       @output = {}
       if block_given?
-        instance_eval(&block)
+        if block.arity == 1
+          yield(self)
+        else
+          instance_eval(&block)
+        end
       end
     end
 
     def output
       CSV.generate(col_sep: "\t") do |tsv|