Rakefile in closure_tree-1.0.0.beta5 vs Rakefile in closure_tree-1.0.0.beta6

@@ -4,20 +4,14 @@
   puts 'You must `gem install bundler` and `bundle install` to run rake tasks'
 end
 
 Bundler::GemHelper.install_tasks
 
+require 'yard'
 
-require 'rdoc/task'
-
-RDoc::Task.new do |rdoc|
-  rdoc.rdoc_dir = 'rdoc'
-  rdoc.title    = 'ClosureTree'
-  rdoc.options << '--line-numbers' << '--inline-source'
-  rdoc.rdoc_files.include('README.rdoc')
-  rdoc.rdoc_files.include('lib/**/*.rb')
+YARD::Rake::YardocTask.new do |t|
+t.files = ['lib/**/*.rb', 'README.md']
 end
-
 
 require 'rake/testtask'
 
 Rake::TestTask.new(:test) do |t|
   t.libs << 'lib'