|
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