lib/ruby-prof/printers/dot_printer.rb in ruby-prof-0.11.2 vs lib/ruby-prof/printers/dot_printer.rb in ruby-prof-0.11.3

@@ -94,10 +94,10 @@
     def print_classes(thread)
       grouped = {}
       thread.methods.each{|m| grouped[m.klass_name] ||= []; grouped[m.klass_name] << m}
       grouped.each do |cls, methods2|
         # Filter down to just seen methods
-        big_methods, small_methods  = methods2.partition{|m| @seen_methods.include? m}
+        big_methods = methods2.select{|m| @seen_methods.include? m}
         
         if !big_methods.empty?
           puts "subgraph cluster_#{cls.object_id} {"
           puts "label = \"#{cls}\";"
           puts "fontcolor = #{CLASS_COLOR};"