lib/polytrix/documentation_generator.rb in polytrix-0.1.0 vs lib/polytrix/documentation_generator.rb in polytrix-0.1.1

@@ -38,10 +38,10 @@
           f.write @result
         end
       end
     end
 
-    def code2doc(source_file, language)
+    def code2doc(source_file, language = nil)
       source_code = File.read(source_file)
       if language.nil?
         language, comment_style = Documentation::CommentStyles.infer File.extname(source_file)
         segmenter_language = comment_style[:language] || language
       else