allow re-entering the root node

fplll · Aug 20, 2018 · 4f568a2 · 4f568a2
1 parent ffb79f9
commit 4f568a2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/fpylll/tools/bkz_stats.py b/src/fpylll/tools/bkz_stats.py
@@ -684,6 +684,9 @@ def reenter(self, **kwds):
 
         """
 
+        if self.current is None:
+            # we exited the root node
+            self.current = self.trace
         node = self.current
         node.data["cputime"]  = node.data.get("cputime",  0) + Accumulator(-time.clock(), repr="sum", count=False)
         node.data["walltime"] = node.data.get("walltime", 0) + Accumulator(-time.time(),  repr="sum", count=False)