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

Integrate docker port mapping, to control non network=host port mapping, including port reassigning for multiple running agents on the same machine

Browse Source
This commit is contained in:
clearml 2025-02-24 13:24:58 +02:00
parent 97cb47d48e
commit d32b82cb01
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
clearml_agent/commands/worker.py
View File

@ -4482,6 +4482,21 @@ class Worker(ServiceCommandSection):
extra_docker_arguments=extra_docker_arguments extra_docker_arguments=extra_docker_arguments
) )
# check if we have mapped ports
try:
port_mappings_struct = DockerArgsSanitizer.resolve_port_mapping(
config=self._session.config, docker_arguments=base_cmd)
if port_mappings_struct:
new_docker_cmd, additional_task_runtime = port_mappings_struct
if new_docker_cmd:
base_cmd = new_docker_cmd
if additional_task_runtime:
from clearml_agent.helper.task_runtime import TaskRuntime
task_runtime = TaskRuntime(self._session)
task_runtime.update_task_runtime(env_task_id, additional_task_runtime)
except Exception as e:
self.log.warning("WARNING: Failed parsing port mapping requested for docker execution: {}".format(e))
# set docker labels # set docker labels
base_cmd += ['-l', self._worker_label.format(worker_id)] base_cmd += ['-l', self._worker_label.format(worker_id)]
base_cmd += ['-l', self._parent_worker_label.format(parent_worker_id)] base_cmd += ['-l', self._parent_worker_label.format(parent_worker_id)]