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_TEMP_STDOUT_FILE_DIR,
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.errors import (
@ -865,10 +869,10 @@ class Worker(ServiceCommandSection):
)
# 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_update_interval = 30.0
_machine_update_interval = ENV_STATUS_REPORT_FREQ_SEC.get() or 30.0
# message printed before starting task logging,
# 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_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):
"""