gem_tasks/treetop.rake in cucumber-0.1.6 vs gem_tasks/treetop.rake in cucumber-0.1.7

@@ -7,15 +7,25 @@
     template = ERB.new(IO.read(File.dirname(__FILE__) + '/../lib/cucumber/treetop_parser/feature.treetop.erb'))
     langs = YAML.load_file(File.dirname(__FILE__) + '/../lib/cucumber/languages.yml')
 
     langs.each do |lang, words|
       grammar_file = File.dirname(__FILE__) + "/../lib/cucumber/treetop_parser/feature_#{lang}.treetop"
+      ruby_file    = File.dirname(__FILE__) + "/../lib/cucumber/treetop_parser/feature_#{lang}.rb"
       grammar = template.result(binding)
       File.open(grammar_file, "wb") do |io|
         io.write(grammar)
       end
       sh "#{tt} #{grammar_file}"
       FileUtils.rm(grammar_file)
+      
+      # Change code so it isn't part of RDoc
+      lines = IO.read(ruby_file).split("\n")
+      lines.each do |line|
+        if line =~ /\s*(def|class|module)/
+          line << " #:nodoc:"
+        end
+      end
+      File.open(ruby_file, 'wb'){|io| io.write(lines.join("\n"))}
     end
   end
 end
 
 namespace :treetop do
\ No newline at end of file