Prevent progress() from raising an exception if you pass in a sequence with less than 10 elements

1 files changed, 2 insertions, 1 deletions

diff --git a/tools.py b/tools.py
index d6b71b1..a884db3 100644
--- a/tools.py
+++ b/tools.py
@@ -13,8 +13,9 @@ def progress(seq):
     n = len(seq)
     print '[' + ' '*21 + ']\r[',
     sys.stdout.flush()
+    update_interval = max(n // 10, 1)
     for i, item in enumerate(seq):
-        if i % (n//10) == 0:
+        if i % update_interval == 0:
             print '.',
             sys.stdout.flush()
         yield item