cyrusyc commited on
Commit
a909029
·
1 Parent(s): 7ab307c

avoid hf loading tasks

Browse files
Files changed (1) hide show
  1. 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
- 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
 
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)