lib/obo_parser/utilities/viz.rb in obo_parser-0.4.1 vs lib/obo_parser/utilities/viz.rb in obo_parser-0.4.2

@@ -12,10 +12,10 @@
         x,y,z = 1,1,1
         center_distance = ( size / 2) 
         edge_length = (size / 2) - (size * 0.2)
         total_terms = ontology.terms.size
         grid_length = Math.cbrt(total_terms).ceil.to_i
-        cutoff ||= size + 1
+        cutoff ||= data.count + 1
 
         data.each_with_index do |row, i|
           break if i > cutoff
           print x * center_distance
           print "\t"