File size: 732 Bytes
bf314e8 | 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 |
from __future__ import annotations
from importlib.metadata import PackageNotFoundError, version
from model.base import HydraModel, HydraModelV2
try:
__version__ = version("onescience.utils.uma")
except PackageNotFoundError:
# package is not installed
__version__ = ""
__all__ = [
"HydraModel",
"HydraModelV2",
"eSCNMDBackbone",
"eSCNMDMoeBackbone",
]
def __getattr__(name: str):
if name == "eSCNMDBackbone":
from model.uma_escn_md import eSCNMDBackbone
return eSCNMDBackbone
if name == "eSCNMDMoeBackbone":
from model.uma_escn_moe import eSCNMDMoeBackbone
return eSCNMDMoeBackbone
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|