allow arbitrary root labels in time tree tracer

fplll · Feb 16, 2018 · deadf0f · deadf0f
1 parent 8880e97
commit deadf0f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/fpylll/tools/bkz_stats.py b/src/fpylll/tools/bkz_stats.py
@@ -632,7 +632,7 @@ class TimeTreeTracer(Tracer):
 
     entries = (("cputime", time.clock), ("walltime", time.time))
 
-    def __init__(self, instance, verbosity=False):
+    def __init__(self, instance, root_label="root", verbosity=False):
         """
         Create a new tracer instance.
 
@@ -641,7 +641,7 @@ def __init__(self, instance, verbosity=False):
 
         """
         Tracer.__init__(self, instance, verbosity)
-        self.trace = Node("root")
+        self.trace = Node(root_label)
         self.current = self.trace
 
     def enter(self, label, **kwds):