emmet-builders 0.84.7rc1__py3-none-any.whl → 0.84.7rc2__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of emmet-builders might be problematic. Click here for more details.
- emmet/builders/materials/ml.py +18 -2
- {emmet_builders-0.84.7rc1.dist-info → emmet_builders-0.84.7rc2.dist-info}/METADATA +1 -1
- {emmet_builders-0.84.7rc1.dist-info → emmet_builders-0.84.7rc2.dist-info}/RECORD +5 -5
- {emmet_builders-0.84.7rc1.dist-info → emmet_builders-0.84.7rc2.dist-info}/WHEEL +1 -1
- {emmet_builders-0.84.7rc1.dist-info → emmet_builders-0.84.7rc2.dist-info}/top_level.txt +0 -0
emmet/builders/materials/ml.py
CHANGED
|
@@ -3,14 +3,26 @@ from typing import TYPE_CHECKING, Optional, Union
|
|
|
3
3
|
|
|
4
4
|
from maggma.builders.map_builder import MapBuilder
|
|
5
5
|
from maggma.core import Store
|
|
6
|
-
|
|
6
|
+
|
|
7
|
+
try:
|
|
8
|
+
from matcalc import PESCalculator
|
|
9
|
+
|
|
10
|
+
matcalc_installed = True
|
|
11
|
+
except ImportError:
|
|
12
|
+
matcalc_installed = False
|
|
13
|
+
|
|
7
14
|
from pymatgen.core import Structure
|
|
8
15
|
|
|
9
16
|
from emmet.core.ml import MLDoc
|
|
10
17
|
from emmet.core.utils import jsanitize
|
|
11
18
|
|
|
12
19
|
if TYPE_CHECKING:
|
|
13
|
-
|
|
20
|
+
try:
|
|
21
|
+
from ase.calculators.calculator import Calculator
|
|
22
|
+
|
|
23
|
+
ase_installed = True
|
|
24
|
+
except ImportError:
|
|
25
|
+
ase_installed = False
|
|
14
26
|
|
|
15
27
|
|
|
16
28
|
class MLBuilder(MapBuilder):
|
|
@@ -40,6 +52,10 @@ class MLBuilder(MapBuilder):
|
|
|
40
52
|
MLDocs. Will be saved in each document so use sparingly. Defaults to None.
|
|
41
53
|
Set to {} to disable default provenance model, version, matcalc_version.
|
|
42
54
|
"""
|
|
55
|
+
|
|
56
|
+
if not matcalc_installed or not ase_installed:
|
|
57
|
+
raise ImportError("Please `pip install matcalc` to use the MLBuilder.")
|
|
58
|
+
|
|
43
59
|
self.materials = materials
|
|
44
60
|
self.ml_potential = ml_potential
|
|
45
61
|
self.kwargs = kwargs
|
|
@@ -18,7 +18,7 @@ emmet/builders/materials/elasticity.py,sha256=Lpp5uE0A3Te_ab_3wFKYON9u05BizkZn7_
|
|
|
18
18
|
emmet/builders/materials/electrodes.py,sha256=wbDB-DpnX_V6xdQXKj7tQzIfq5JvqfJDw78QPNDVLDo,23768
|
|
19
19
|
emmet/builders/materials/electronic_structure.py,sha256=5jFPJOuF3OQ_CrpD8B0kkHcd3b0ZwCLoerbMvUYmP1g,29421
|
|
20
20
|
emmet/builders/materials/magnetism.py,sha256=JRy2ljJwNuzCybeJIp9oyfL5b70cCViwGFJDY2RDxuA,4605
|
|
21
|
-
emmet/builders/materials/ml.py,sha256=
|
|
21
|
+
emmet/builders/materials/ml.py,sha256=9IGHJUjwRMrzwXp0_218QS-LAmXEoqmyFqWoxY3UmZM,3599
|
|
22
22
|
emmet/builders/materials/optimade.py,sha256=I2JU2rdyqFVCD3QxiwK6N4B3R5JfK_OZiF3DebYfxBo,5164
|
|
23
23
|
emmet/builders/materials/oxidation_states.py,sha256=ptTfakXk1bjexniBzx87XCutiuRiolFpnMZETsJlj6I,1670
|
|
24
24
|
emmet/builders/materials/piezoelectric.py,sha256=pm7wqlynIZwmDNIcM6EiL1A_t7G5UVri2o59hKkxcfg,7726
|
|
@@ -48,7 +48,7 @@ emmet/builders/vasp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
|
|
|
48
48
|
emmet/builders/vasp/materials.py,sha256=PAqx-_-Z7y4uw87SPJuP9mgeZXIlZe0A2WiA47VFZWw,12929
|
|
49
49
|
emmet/builders/vasp/mp_potcar_stats.json.gz,sha256=x3bn4gSMj1U_3bR2qKIaBtbJlYT-EJgoUIMFTA9bvaU,338957
|
|
50
50
|
emmet/builders/vasp/task_validator.py,sha256=bmRTDiOWof4rpHVg3ksoxocN9xxieYu7IE-ylMjYOVs,2922
|
|
51
|
-
emmet_builders-0.84.
|
|
52
|
-
emmet_builders-0.84.
|
|
53
|
-
emmet_builders-0.84.
|
|
54
|
-
emmet_builders-0.84.
|
|
51
|
+
emmet_builders-0.84.7rc2.dist-info/METADATA,sha256=YXt1baqBFv0pucAGgaOKSth3W6mQ1KPi4pT5N2rCG1I,2361
|
|
52
|
+
emmet_builders-0.84.7rc2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
53
|
+
emmet_builders-0.84.7rc2.dist-info/top_level.txt,sha256=6GcpbmWPeFhNCTfDFilb8GQ4T1UQu4z9c5jpobjwE-Q,6
|
|
54
|
+
emmet_builders-0.84.7rc2.dist-info/RECORD,,
|
|
File without changes
|