File size: 890 Bytes
c2767f4 | 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 30 31 32 33 34 35 | __all__ = [
"CGModel",
"build_score_model",
"load_model_args",
"load_score_model",
"model_uses_lm_embeddings",
]
def __getattr__(name):
if name == "CGModel":
from .cg_model import CGModel
return CGModel
if name in {
"build_score_model",
"load_model_args",
"load_score_model",
"model_uses_lm_embeddings",
}:
from .score_wrapper import (
build_score_model,
load_model_args,
load_score_model,
model_uses_lm_embeddings,
)
return {
"build_score_model": build_score_model,
"load_model_args": load_model_args,
"load_score_model": load_score_model,
"model_uses_lm_embeddings": model_uses_lm_embeddings,
}[name]
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|