Prevent ZeroDivisionError when options.colour_nodes_by_selftime is set

StOnEGiggity · Jun 11, 2016 · 7f295e1 · 7f295e1
1 parent 2f3c164
commit 7f295e1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gprof2dot.py b/gprof2dot.py
@@ -714,13 +714,13 @@ def prune(self, node_thres, edge_thres, colour_nodes_by_selftime):
                     weights.append(function[TIME_RATIO])
                 except UndefinedEvent:
                     pass
-            max_ratio = max(weights)
+            max_ratio = max(weights or [1])
 
             # apply rescaled weights for coloriung
             for function in compat_itervalues(self.functions):
                 try:
                     function.weight = function[TIME_RATIO] / max_ratio
-                except UndefinedEvent:
+                except (ZeroDivisionError, UndefinedEvent):
                     pass
 
     def dump(self):