Преглед изворни кода

utils.debug: Fixes memsample IndexError bug

Ask Solem пре 11 година
родитељ
комит
894023a414
1 измењених фајлова са 4 додато и 1 уклоњено
  1. 4 1
      celery/utils/debug.py

+ 4 - 1
celery/utils/debug.py

@@ -117,7 +117,10 @@ def sample(x, n, k=0):
     """
     j = len(x) // n
     for _ in range(n):
-        yield x[k]
+        try:
+            yield x[k]
+        except IndexError:
+            break
         k += j