Fix queue handling for backwards compatibility

This commit is contained in:
allegroai 2024-03-17 19:00:18 +02:00
parent 56058beec2
commit 10fb157d58

View File

@ -1095,8 +1095,9 @@ class K8sIntegration(Worker):
:param list(str) queue: queue name to pull from :param list(str) queue: queue name to pull from
""" """
queues = queue if isinstance(queue, (list, tuple)) else ([queue] if queue else None)
return self.daemon( return self.daemon(
queues=[ObjectID(name=q) for q in queue] if queue else None, queues=[ObjectID(name=q) for q in queues] if queues else None,
log_level=logging.INFO, foreground=True, docker=False, **kwargs, log_level=logging.INFO, foreground=True, docker=False, **kwargs,
) )