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

Add multiple packages support

Browse Source
This commit is contained in:
allegroai 2020-12-10 09:51:49 +02:00
parent 259113c989
commit a41ea52f87
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
trains_agent/commands/worker.py
View File

@ -323,6 +323,7 @@ class Worker(ServiceCommandSection):
PostRequirement, PostRequirement,
ExternalRequirements, ExternalRequirements,
partial(PackageCollectorRequirement, collect_package=['trains']), partial(PackageCollectorRequirement, collect_package=['trains']),
partial(PackageCollectorRequirement, collect_package=['clearml']),
) )
# poll queues every _polling_interval seconds # poll queues every _polling_interval seconds

4
trains_agent/helper/package/priority_req.py
View File

@ -49,12 +49,12 @@ class PackageCollectorRequirement(SimpleSubstitution):
def __init__(self, session, collect_package): def __init__(self, session, collect_package):
super(PackageCollectorRequirement, self).__init__(session) super(PackageCollectorRequirement, self).__init__(session)
self._collect_packages = collect_package self._collect_packages = collect_package or tuple()
self._last_req = None self._last_req = None
def match(self, req): def match(self, req):
# match package names # match package names
return req.name and (req.name.lower() in self.name or req.name.lower() in self._collect_packages) return req.name and req.name.lower() in self._collect_packages
def replace(self, req): def replace(self, req):
""" """