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

Add support for extra fields in request object

Browse Source
This commit is contained in:
allegroai 2021-07-20 10:06:05 +03:00
parent fb6fd9ac4a
commit cf9671c253
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
clearml/backend_api/session/request.py
View File

@ -11,8 +11,19 @@ class Request(ApiModel):
_method = 'get' _method = 'get'
def __init__(self, **kwargs): def __init__(self, **kwargs):
if kwargs: allow_extra_fields = kwargs.pop("_allow_extra_fields_", False)
if not allow_extra_fields and kwargs:
raise ValueError('Unsupported keyword arguments: %s' % ', '.join(kwargs.keys())) raise ValueError('Unsupported keyword arguments: %s' % ', '.join(kwargs.keys()))
elif allow_extra_fields and kwargs:
self._extra_fields = kwargs
else:
self._extra_fields = {}
def to_dict(self, *args, **kwargs):
res = super(Request, self).to_dict(*args, **kwargs)
if self._extra_fields:
res.update(self._extra_fields)
return res
@six.add_metaclass(abc.ABCMeta) @six.add_metaclass(abc.ABCMeta)