diff --git a/Interlace/lib/threader.py b/Interlace/lib/threader.py index 2bc7af7..1038eaa 100644 --- a/Interlace/lib/threader.py +++ b/Interlace/lib/threader.py @@ -66,17 +66,17 @@ class Worker(object): self.tqdm = tqdm def __call__(self): + queue = self.queue while True: try: - # get task from queue - task = next(self.queue) + task = next(queue) if isinstance(self.tqdm, tqdm): self.tqdm.update(1) # run task task.run(self.tqdm) else: task.run() - except IndexError: + except StopIteration: break