fix pymatgen import error

yoshida-lab · Mar 14, 2021 · 5407b04 · 5407b04
1 parent 8af93eb
commit 5407b04
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/xenonpy/descriptor/structure.py b/xenonpy/descriptor/structure.py
@@ -5,7 +5,7 @@
 import re
 
 import numpy as np
-import pymatgen as pm
+from pymatgen.core import Element
 from pymatgen.analysis.local_env import VoronoiNN
 
 from xenonpy.descriptor.base import BaseDescriptor, BaseFeaturizer
@@ -165,7 +165,7 @@ def get_element_representation(name):
         name: string
             element symbol
         """
-        element = pm.Element(name)
+        element = Element(name)
         general_element_electronic = {
             's1': 0.0,
             's2': 0.0,