__init__.py

 
"""
This package encapsulates different PXD background generator models.
"""
 
from importlib import import_module
from typing import Callable, Type
 
 
 
def _get_model_cls(model: str) -> Type:
    cls = None
    try:
        cls = import_module(f".{model}", __name__).Model
    except AttributeError as exc:
        exc.msg = f"please define a model class for {model!r}"
        raise
    except ModuleNotFoundError as exc:
        exc.msg = f"the selected model {model!r} is not implemented"
        raise
    return cls
 
 
 
def _get_generate_func(model: str) -> Callable:
    func = None
    try:
        func = import_module(f".{model}", __name__).generate
    except AttributeError as exc:
        exc.msg = f"please define the generation function for {model!r}"
        raise
    except ModuleNotFoundError as exc:
        exc.msg = f"the selected model {model!r} is not implemented"
        raise
    return func
 
 
 
MODELS = (
    "convnet",
    "resnet",
)