Spaces:
Running
Running
avoid hf loading tasks
Browse files- mlip_arena/tasks/__init__.py +10 -8
mlip_arena/tasks/__init__.py
CHANGED
@@ -6,15 +6,17 @@ from huggingface_hub import HfApi, HfFileSystem, hf_hub_download
|
|
6 |
from mlip_arena.models import MLIP
|
7 |
from mlip_arena.models import REGISTRY as MODEL_REGISTRY
|
8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
from .neb import run as NEB
|
14 |
-
from .neb import run_from_endpoints as NEB_FROM_ENDPOINTS
|
15 |
-
from .optimize import run as OPT
|
16 |
-
|
17 |
-
__all__ = ["OPT", "EOS", "MD", "NEB", "NEB_FROM_ENDPOINTS", "ELASTICITY"]
|
18 |
|
19 |
with open(Path(__file__).parent / "registry.yaml", encoding="utf-8") as f:
|
20 |
REGISTRY = yaml.safe_load(f)
|
|
|
6 |
from mlip_arena.models import MLIP
|
7 |
from mlip_arena.models import REGISTRY as MODEL_REGISTRY
|
8 |
|
9 |
+
try:
|
10 |
+
from .elasticity import run as ELASTICITY
|
11 |
+
from .eos import run as EOS
|
12 |
+
from .md import run as MD
|
13 |
+
from .neb import run as NEB
|
14 |
+
from .neb import run_from_endpoints as NEB_FROM_ENDPOINTS
|
15 |
+
from .optimize import run as OPT
|
16 |
|
17 |
+
__all__ = ["OPT", "EOS", "MD", "NEB", "NEB_FROM_ENDPOINTS", "ELASTICITY"]
|
18 |
+
except ImportError:
|
19 |
+
pass
|
|
|
|
|
|
|
|
|
|
|
20 |
|
21 |
with open(Path(__file__).parent / "registry.yaml", encoding="utf-8") as f:
|
22 |
REGISTRY = yaml.safe_load(f)
|