From 1f871cbad5b2eac1a17db90da83d2ca6bd711ee8 Mon Sep 17 00:00:00 2001
From: Graham Conzett <gconzett@covermymeds.com>
Date: Mon, 3 Feb 2014 14:30:05 -0500
Subject: [PATCH] Allow passing through file extensions to graph

Still defaults to png.
---
 lib/decisiontree/id3_tree.rb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lib/decisiontree/id3_tree.rb b/lib/decisiontree/id3_tree.rb
index 04c9191..afeceb1 100755
--- a/lib/decisiontree/id3_tree.rb
+++ b/lib/decisiontree/id3_tree.rb
@@ -131,10 +131,10 @@ module DecisionTree
       descend(@tree, test)
     end
 
-    def graph(filename)
+    def graph(filename, file_type = "png")
       require 'graphr'
       dgp = DotGraphPrinter.new(build_tree)
-      dgp.write_to_file("#{filename}.png", "png")
+      dgp.write_to_file("#{filename}.#{file_type}", file_type)
     end
 
     def ruleset