ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-06-26 18:16:15 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add CLEARML_AGENT_QUEUE_POLL_FREQ_SEC and CLEARML_AGENT_STATUS_REPORT_FREQ_SEC env vars to customize agent behavior

Browse Source
This commit is contained in:
clearml 2025-06-22 22:38:44 +03:00
parent 8d8dc4e396
commit afe69c822f
2 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
clearml_agent/commands/worker.py
View File

@ -79,7 +79,11 @@ from clearml_agent.definitions import (
ENV_AGENT_FORCE_EXEC_SCRIPT, ENV_AGENT_FORCE_EXEC_SCRIPT,
ENV_TEMP_STDOUT_FILE_DIR, ENV_TEMP_STDOUT_FILE_DIR,
ENV_AGENT_FORCE_TASK_INIT, ENV_AGENT_FORCE_TASK_INIT,
ENV_AGENT_DEBUG_GET_NEXT_TASK, ENV_ABORT_CALLBACK_CMD, ENV_ABORT_CALLBACK_CMD_TIMEOUT, ENV_AGENT_DEBUG_GET_NEXT_TASK,
ENV_ABORT_CALLBACK_CMD,
ENV_ABORT_CALLBACK_CMD_TIMEOUT,
ENV_QUEUE_POLL_FREQ_SEC,
ENV_STATUS_REPORT_FREQ_SEC,
) )
from clearml_agent.definitions import WORKING_REPOSITORY_DIR, PIP_EXTRA_INDICES from clearml_agent.definitions import WORKING_REPOSITORY_DIR, PIP_EXTRA_INDICES
from clearml_agent.errors import ( from clearml_agent.errors import (
@ -865,10 +869,10 @@ class Worker(ServiceCommandSection):
) )
# default poll queues every _polling_interval seconds # default poll queues every _polling_interval seconds
_polling_interval = 5.0 _polling_interval = ENV_QUEUE_POLL_FREQ_SEC.get() or 5.0
# machine status update intervals, seconds # machine status update intervals, seconds
_machine_update_interval = 30.0 _machine_update_interval = ENV_STATUS_REPORT_FREQ_SEC.get() or 30.0
# message printed before starting task logging, # message printed before starting task logging,
# it will be parsed by services_mode, to identify internal docker logging start # it will be parsed by services_mode, to identify internal docker logging start

4
clearml_agent/definitions.py
View File

@ -256,6 +256,10 @@ ENV_GIT_CLONE_VERBOSE = EnvironmentConfig("CLEARML_AGENT_GIT_CLONE_VERBOSE", typ
ENV_GPU_FRACTIONS = EnvironmentConfig("CLEARML_AGENT_GPU_FRACTIONS") ENV_GPU_FRACTIONS = EnvironmentConfig("CLEARML_AGENT_GPU_FRACTIONS")
ENV_QUEUE_POLL_FREQ_SEC = EnvironmentConfig("CLEARML_AGENT_QUEUE_POLL_FREQ_SEC", type=float)
ENV_STATUS_REPORT_FREQ_SEC = EnvironmentConfig("CLEARML_AGENT_STATUS_REPORT_FREQ_SEC", type=float)
class FileBuffering(IntEnum): class FileBuffering(IntEnum):
""" """