From 3248c93f2fddf9b17148273af5dd5e2af2cb537b Mon Sep 17 00:00:00 2001
From: Luke Kanies <luke@madstop.com>
Date: Tue, 11 Dec 2007 00:07:20 -0600
Subject: Fixing #937 -- I had not ported the dot methods at all, and I had to
 make a few small changes to make them work.

---
 lib/puppet/simple_graph.rb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'lib')

diff --git a/lib/puppet/simple_graph.rb b/lib/puppet/simple_graph.rb
index 48bf4991e..c9920e60a 100644
--- a/lib/puppet/simple_graph.rb
+++ b/lib/puppet/simple_graph.rb
@@ -212,7 +212,7 @@ class Puppet::SimpleGraph
         params = {'name'     => '"'+name+'"',
                   'fontsize' => fontsize,
                   'label'    => name}
-        v_label = vertex_label(v)
+        v_label = v.to_s
         params.merge!(v_label) if v_label and v_label.kind_of? Hash
         graph << DOT::DOTNode.new(params)
       end
@@ -220,7 +220,7 @@ class Puppet::SimpleGraph
         params = {'from'     => '"'+ e.source.to_s + '"',
                   'to'       => '"'+ e.target.to_s + '"',
                   'fontsize' => fontsize }
-        e_label = edge_label(e)
+        e_label = e.to_s
         params.merge!(e_label) if e_label and e_label.kind_of? Hash
         graph << edge_klass.new(params)
       end
-- 
cgit