update

CederGroupHub · Jan 22, 2021 · 16b9383 · 16b9383
1 parent 9b68459
commit 16b9383
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/materials_entity_recognition/scripts/sent_ele_func.py b/materials_entity_recognition/scripts/sent_ele_func.py
@@ -1,12 +1,14 @@
-from material_parser.material_parser import MaterialParser
-import  regex
+from .utils import found_package
+import regex
 
 __author__ = 'Tanjin He'
 __maintainer__ = 'Tanjin He, Ziqin (Shaun) Rong'
 __email__ = '[email protected], [email protected]'
 
 # constant
-mp = MaterialParser(pubchem_lookup=False)
+if found_package('material_parser'):
+    from material_parser.material_parser import MaterialParser
+    mp = MaterialParser(pubchem_lookup=False)
 allNonMetalElements = set(['C', 'H', 'O', 'N', 'Cl', 'F', 'P', 'S', 'Br', 'I', 'Se'] + ['He', 'Ne', 'Ar', 'Kr', 'Xe', 'Rn'])
 # element table by symbol of elements
 elementTable = {