|
21 | 21 | LightStructureEnvironments, |
22 | 22 | ) |
23 | 23 | from pymatgen.analysis.graphs import MoleculeGraph, StructureGraph |
24 | | -from pymatgen.analysis.local_env import LocalStructOrderParams, cn_opt_params |
| 24 | +from pymatgen.analysis.local_env import LocalStructOrderParams, CN_OPT_PARAMS |
25 | 25 | from pymatgen.core import Molecule, Structure |
26 | 26 | from pymatgen.ext.matproj import MPRester |
27 | 27 | from pymatgen.io.ase import AseAtomsAdaptor |
@@ -77,14 +77,14 @@ def _get_local_order_parameters(structure_graph, n): |
77 | 77 | # code from @nisse3000, moved here from graphs to avoid circular |
78 | 78 | # import, also makes sense to have this as a general NN method |
79 | 79 | cn = structure_graph.get_coordination_of_site(n) |
80 | | - if cn in [int(k_cn) for k_cn in cn_opt_params]: |
81 | | - names = list(cn_opt_params[cn]) |
| 80 | + if cn in [int(k_cn) for k_cn in CN_OPT_PARAMS]: |
| 81 | + names = list(CN_OPT_PARAMS[cn]) |
82 | 82 | types = [] |
83 | 83 | params = [] |
84 | 84 | for name in names: |
85 | | - types.append(cn_opt_params[cn][name][0]) |
| 85 | + types.append(CN_OPT_PARAMS[cn][name][0]) |
86 | 86 | tmp = ( |
87 | | - cn_opt_params[cn][name][1] if len(cn_opt_params[cn][name]) > 1 else None |
| 87 | + CN_OPT_PARAMS[cn][name][1] if len(CN_OPT_PARAMS[cn][name]) > 1 else None |
88 | 88 | ) |
89 | 89 | params.append(tmp) |
90 | 90 | lost_ops = LocalStructOrderParams(types, parameters=params) |
|
0 commit comments