skip TestInsertionWorkflow if ChargeInsertionAnalyzer not importable

following @jmmshn's recommendation, see materialsproject/pymatgen#2582 (comment)

Author: janosh

atomate/vasp/firetasks/electrode_tasks.py

@@ -1,7 +1,6 @@
 import math
 
 from fireworks import FiretaskBase, Firework, FWAction, Workflow, explicit_serialize
-from pymatgen.analysis.defects.utils import ChargeInsertionAnalyzer
 from pymatgen.analysis.structure_matcher import StructureMatcher
 from pymatgen.core import Structure
 
@@ -12,6 +11,15 @@
 from atomate.vasp.firetasks import pass_vasp_result
 from atomate.vasp.fireworks.core import OptimizeFW, StaticFW
 
+try:
+    from pymatgen.analysis.defects import ChargeInsertionAnalyzer
+except ImportError:
+    print(
+        "Failed to import ChargeInsertionAnalyzer. This is likely due to converting the pymatgen defects module "
+        "to a namespace package. See https://github.com/materialsproject/pymatgen/pull/2582#issuecomment-1198318101 "
+        "for updates."
+    )
+
 __author__ = "Jimmy Shen"
 __email__ = "[email protected]"
 

atomate/vasp/workflows/tests/test_insertion_workflow.py

@@ -1,4 +1,5 @@
 import os
+import unittest
 from pathlib import Path
 
 from fireworks.core.fworker import FWorker
@@ -10,6 +11,12 @@
 from atomate.vasp.powerups import add_modify_incar, use_fake_vasp, use_potcar_spec
 from atomate.vasp.workflows.base.electrode import get_ion_insertion_wf
 
+try:
+    from pymatgen.analysis.defects import ChargeInsertionAnalyzer
+except ImportError:
+    ChargeInsertionAnalyzer = None
+
+
 __author__ = "Jimmy Shen"
 __email__ = "[email protected]"
 
@@ -19,6 +26,9 @@
 wf_dir = ref_dir / "insertion_wf"
 
 
+@unittest.skipIf(
+    ChargeInsertionAnalyzer is None, "pymatgen.analysis.defects not installed"
+)
 class TestInsertionWorkflow(AtomateTest):
     def setUp(self):
         super().setUp()