count restarts

vecto-ai · Jul 5, 2021 · e116b1b · e116b1b
1 parent 357d218
commit e116b1b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/vecto/corpus/iterators.py b/vecto/corpus/iterators.py
@@ -104,6 +104,7 @@ def __init__(self, tree, start):
         self.tree = tree
         self.id_file = start[0]
         self.start_offset = start[1]
+        self._cnt_restarts = 0
 
     def _generate_samples(self):
         filename = self.tree[self.id_file][0]
@@ -117,6 +118,7 @@ def _generate_samples(self):
             self.id_file += 1
             if self.id_file >= len(self.tree):
                 self.id_file = 0
+                self._cnt_restarts += 1
             file_in = detect_archive_format_and_open(self.tree[self.id_file][0])