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

Add worker command state enforcement conforming and verification callback

Browse Source
This commit is contained in:
allegroai 2020-05-09 19:42:51 +03:00
parent 06bfea80bc
commit 38e02ca5cd
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
trains_agent/commands/base.py
View File

@ -94,9 +94,20 @@ class ServiceCommandSection(BaseCommandSection):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super(ServiceCommandSection, self).__init__() super(ServiceCommandSection, self).__init__()
kwargs = self._verify_command_states(kwargs)
self._session = self._get_session(*args, **kwargs) self._session = self._get_session(*args, **kwargs)
self._list_formatter = ListFormatter(self.service) self._list_formatter = ListFormatter(self.service)
@classmethod
def _verify_command_states(cls, kwargs):
"""
Conform and enforce command argument
This is where you can automatically turn on/off switches based on different states.
:param kwargs:
:return: kwargs
"""
return kwargs
@staticmethod @staticmethod
def _get_session(*args, **kwargs): def _get_session(*args, **kwargs):
return Session(*args, **kwargs) return Session(*args, **kwargs)