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

Refactor APIClient

Browse Source
This commit is contained in:
allegroai 2020-10-30 09:49:55 +02:00
parent 56db3c67e7
commit a3e53a727f
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
trains/backend_api/session/client/client.py
View File

@ -2,6 +2,7 @@ from __future__ import unicode_literals
import abc
import os
import types
from argparse import Namespace
from collections import OrderedDict
from enum import Enum
@ -520,9 +521,11 @@ class APIClient(object):
models = None # type: Any
projects = None # type: Any
def __init__(self, session=None, api_version=None):
def __init__(self, session=None, api_version=None, **kwargs):
self.session = session or StrictSession()
_api_services = kwargs.pop("api_services", api_services)
def import_(*args, **kwargs):
try:
return import_module(*args, **kwargs)
@ -536,16 +539,20 @@ class APIClient(object):
for name, mod in (
(
name,
import_(".".join((api_services.__name__, api_version, name))),
import_(".".join((_api_services.__name__, api_version, name))),
)
for name in api_services.__all__
for name in _api_services.__all__
)
if mod
)
else:
services = OrderedDict(
(name, getattr(api_services, name)) for name in api_services.__all__
(name, getattr(_api_services, name)) for name in _api_services.__all__
)
self._update_services(services)
def _update_services(self, services):
# type: (Dict[str, types.ModuleType]) -> ()
self.__dict__.update(
dict(
{
@ -553,4 +560,4 @@ class APIClient(object):
for name, module in services.items()
},
)
)
)