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

Fix restart feature should be tested against agent session

Browse Source
This commit is contained in:
allegroai 2023-03-28 14:33:33 +03:00
parent 40456be948
commit 46ded2864d
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clearml_agent/commands/worker.py
View File

@ -736,9 +736,9 @@ class Worker(ServiceCommandSection):
except Exception:
pass
def _get_docker_restart_value(self, session, task_id: str):
def _get_docker_restart_value(self, task_session, task_id: str):
try:
session.verify_feature_set('advanced')
self._session.verify_feature_set('advanced')
except ValueError:
return
@ -760,7 +760,7 @@ class Worker(ServiceCommandSection):
path, _, expected_value = selector.partition("=")
result = session.send_request(
result = task_session.send_request(
service='tasks',
action='get_all',
json={'id': [task_id], 'only_fields': [path], 'search_hidden': True},