cli: protect task_type import error

materialsproject · Mar 3, 2021 · 89801d1 · 89801d1
1 parent 5a1a036
commit 89801d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/emmet-cli/emmet/cli/calc.py b/emmet-cli/emmet/cli/calc.py
@@ -15,7 +15,6 @@
 from pymatgen.util.provenance import StructureNL, Author
 
 from emmet.core.utils import group_structures, get_sg
-from emmet.core.vasp.calc_types import task_type
 from emmet.cli import SETTINGS
 from emmet.cli.utils import calcdb_from_mgrant, aggregate_by_formula, structures_match
 from emmet.cli.utils import get_meta_from_structure, load_structure
@@ -37,6 +36,7 @@ def get_format(fname):
 
 
 def load_canonical_structures(ctx, full_name, formula):
+    from emmet.core.vasp.calc_types import task_type  # TODO import error
     collection = ctx.obj["COLLECTIONS"][full_name]
 
     if formula not in canonical_structures[full_name]: