ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-03-03 10:42:05 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Allow providing queue names to daemon

Browse Source
This commit is contained in:
allegroai 2020-02-26 16:58:25 +02:00
parent 596093aac6
commit f838c8fc70
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
trains_agent/commands/worker.py
View File

@ -631,11 +631,13 @@ class Worker(ServiceCommandSection):
self.log.debug("starting resource monitor thread") self.log.debug("starting resource monitor thread")
print("Worker \"{}\" - ".format(self.worker_id), end='') print("Worker \"{}\" - ".format(self.worker_id), end='')
if not queues: if queues:
queues = return_list(queues)
queues = [self._resolve_name(q, "queues") for q in queues]
else:
default_queue = self._session.send_api(queues_api.GetDefaultRequest()) default_queue = self._session.send_api(queues_api.GetDefaultRequest())
queues = [default_queue.id] queues = [default_queue.id]
queues = return_list(queues)
queues_info = [ queues_info = [
self._session.send_api( self._session.send_api(
queues_api.GetByIdRequest(queue) queues_api.GetByIdRequest(queue)