diff --git a/utils/queues.py b/utils/queues.py index 85a4e63..84338fd 100644 --- a/utils/queues.py +++ b/utils/queues.py @@ -13,9 +13,8 @@ class TasksHandlerMixin: def poll(self): while True: response: tasks_pb2.TakeResponse = self.stub.Take(tasks_pb2.TakeRequest(queue=self.queue_name)) - task: tasks_pb2.Task = response.task - print(task) - if not task: + task = response.task + if not task.id: time.sleep(0.2) continue try: