44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
|
|
import inspect
|
|
|
|
from scepter.modules.utils.registry import Registry, deep_copy
|
|
|
|
|
|
def build_solver(cfg, registry, logger=None, *args, **kwargs):
|
|
from scepter.modules.utils.config import Config
|
|
if not isinstance(cfg, Config):
|
|
raise TypeError(f'config must be type Config, got {type(cfg)}')
|
|
if not cfg.have('NAME'):
|
|
raise KeyError(f'config must contain key NAME, got {cfg}')
|
|
if not isinstance(registry, Registry):
|
|
raise TypeError(
|
|
f'registry must be type Registry, got {type(registry)}')
|
|
|
|
cfg = deep_copy(cfg)
|
|
req_type = cfg.get('NAME')
|
|
|
|
from scepter.modules.utils.import_utils import LazyImportModule
|
|
sig = (registry.name.upper(), req_type)
|
|
LazyImportModule.import_module(sig)
|
|
if isinstance(req_type, str):
|
|
req_type_entry = registry.get(req_type)
|
|
if req_type_entry is None:
|
|
raise KeyError(f'{req_type} not found in {registry.name} registry')
|
|
|
|
if kwargs is not None:
|
|
cfg._update_dict(kwargs)
|
|
|
|
if inspect.isclass(req_type_entry):
|
|
try:
|
|
return req_type_entry(cfg, logger=logger, *args, **kwargs)
|
|
except Exception as e:
|
|
raise Exception(f'Failed to init class {req_type_entry}, with {e}')
|
|
else:
|
|
raise TypeError(
|
|
f'type must be str or class, got {type(req_type_entry)}')
|
|
|
|
|
|
SOLVERS = Registry('SOLVERS', build_func=build_solver, allow_types=('class', ))
|