fix ImportError: cannot import pymatgen.symmetry.analyzer.SymmetryUnd…

…eterminedError
janosh · Oct 18, 2024 · 298c49a · 298c49a
1 parent 74b315b
commit 298c49a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/pymatviz/structure_viz/mpl.py b/pymatviz/structure_viz/mpl.py
@@ -19,7 +19,7 @@
 from matplotlib.path import Path
 from pymatgen.analysis.local_env import CrystalNN, NearNeighbors
 from pymatgen.core import Structure
-from pymatgen.symmetry.analyzer import SpacegroupAnalyzer, SymmetryUndeterminedError
+from pymatgen.symmetry.analyzer import SpacegroupAnalyzer
 
 from pymatviz.colors import ELEM_COLORS_JMOL, ELEM_COLORS_VESTA
 from pymatviz.enums import ElemColorScheme, Key
@@ -199,7 +199,7 @@ class used to plot chemical bonds. Allowed are edgecolor, facecolor, color,
             try:
                 spg_analyzer = SpacegroupAnalyzer(struct)
                 struct = spg_analyzer.get_conventional_standard_structure()
-            except SymmetryUndeterminedError:
+            except ValueError:
                 warnings.warn(NO_SYM_MSG, UserWarning, stacklevel=2)
 
         # Get default colors

diff --git a/pymatviz/structure_viz/plotly.py b/pymatviz/structure_viz/plotly.py
@@ -11,7 +11,7 @@
 import plotly.graph_objects as go
 from plotly.subplots import make_subplots
 from pymatgen.analysis.local_env import CrystalNN, NearNeighbors
-from pymatgen.symmetry.analyzer import SpacegroupAnalyzer, SymmetryUndeterminedError
+from pymatgen.symmetry.analyzer import SpacegroupAnalyzer
 
 from pymatviz.enums import ElemColorScheme, SiteCoords
 from pymatviz.structure_viz.helpers import (
@@ -163,7 +163,7 @@ def structure_2d_plotly(
             try:
                 spg_analyzer = SpacegroupAnalyzer(struct_i)
                 struct_i = spg_analyzer.get_conventional_standard_structure()  # noqa: PLW2901
-            except SymmetryUndeterminedError:
+            except ValueError:
                 warnings.warn(NO_SYM_MSG, UserWarning, stacklevel=2)
 
         # Apply rotation
@@ -429,7 +429,7 @@ def structure_3d_plotly(
             try:
                 spg_analyzer = SpacegroupAnalyzer(struct_i)
                 struct_i = spg_analyzer.get_conventional_standard_structure()  # noqa: PLW2901
-            except SymmetryUndeterminedError:
+            except ValueError:
                 warnings.warn(NO_SYM_MSG, UserWarning, stacklevel=2)
 
         visible_image_atoms: set[tuple[float, float, float]] = set()