From a909029b70b3e35138a17b1b072c13e7d050c48c Mon Sep 17 00:00:00 2001 From: Yuan Chiang Date: Sun, 12 Jan 2025 16:22:30 -0800 Subject: [PATCH] avoid hf loading tasks --- mlip_arena/tasks/__init__.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/mlip_arena/tasks/__init__.py b/mlip_arena/tasks/__init__.py index bdaf2e9..733c309 100644 --- a/mlip_arena/tasks/__init__.py +++ b/mlip_arena/tasks/__init__.py @@ -6,15 +6,17 @@ from mlip_arena.models import MLIP from mlip_arena.models import REGISTRY as MODEL_REGISTRY +try: + from .elasticity import run as ELASTICITY + from .eos import run as EOS + from .md import run as MD + from .neb import run as NEB + from .neb import run_from_endpoints as NEB_FROM_ENDPOINTS + from .optimize import run as OPT -from .elasticity import run as ELASTICITY -from .eos import run as EOS -from .md import run as MD -from .neb import run as NEB -from .neb import run_from_endpoints as NEB_FROM_ENDPOINTS -from .optimize import run as OPT - -__all__ = ["OPT", "EOS", "MD", "NEB", "NEB_FROM_ENDPOINTS", "ELASTICITY"] + __all__ = ["OPT", "EOS", "MD", "NEB", "NEB_FROM_ENDPOINTS", "ELASTICITY"] +except ImportError: + pass with open(Path(__file__).parent / "registry.yaml", encoding="utf-8") as f: REGISTRY = yaml.safe_load(f)