Rakefile in git-commit-notifier-0.10.4 vs Rakefile in git-commit-notifier-0.10.5

@@ -24,17 +24,18 @@
   $stderr.puts "RSpec not available. Install it with: gem install rspec-core rspec-expectations"
 end
 
 task :default => :spec
 
-require 'rake/rdoctask'
-Rake::RDocTask.new do |rdoc|
-  version = File.exists?('VERSION') ? IO.read('VERSION') : ""
+begin
+  require 'yard'
 
-  rdoc.rdoc_dir = 'rdoc'
-  rdoc.title = "git-commit-notifier #{version}"
-  rdoc.rdoc_files.include('README*')
-  rdoc.rdoc_files.include('lib/**/*.rb')
+  YARD::Rake::YardocTask.new do |yard|
+    version = File.exists?('VERSION') ? IO.read('VERSION') : ""
+    yard.options << "--title='git-commit-notifier #{version}'"
+  end
+rescue LoadError
+  $stderr.puts "Please install YARD with: gem install yard"
 end
 
 begin
   gem 'code-cleaner'
   desc 'Clean code for whitespaces and tabs'