File size: 804 Bytes
205a7af |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
import importlib.util
from siclib.models.base_model import BaseModel
from siclib.utils.tools import get_class
def get_model(name):
import_paths = [
name,
f"{__name__}.{name}",
]
for path in import_paths:
try:
spec = importlib.util.find_spec(path)
except ModuleNotFoundError:
spec = None
if spec is not None:
try:
return get_class(path, BaseModel)
except AssertionError:
mod = __import__(path, fromlist=[""])
try:
return mod.__main_model__
except AttributeError as exc:
print(exc)
continue
raise RuntimeError(f'Model {name} not found in any of [{" ".join(import_paths)}]')
|