add optional dep seekpath under new dep group 'brillouin'

janosh · Nov 29, 2024 · 1f00492 · 1f00492
1 parent 15c2642
commit 1f00492
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pymatviz/brillouin.py b/pymatviz/brillouin.py
@@ -183,9 +183,10 @@ def brillouin_zone_3d(
             x_coords += [coords[0]]
             y_coords += [coords[1]]
             z_coords += [coords[2]]
-            pretty_label = label.replace("\\Gamma", "Γ").replace("GAMMA", "Γ")
+            for text, repl in [("\\Gamma", "Γ"), ("GAMMA", "Γ"), ("DELTA", "Δ")]:
+                label = label.replace(text, repl)  # noqa: PLW2901
             # use <sub> for subscripts
-            pretty_label = re.sub(r"_(\d+)", r"<sub>\1</sub>", pretty_label)
+            pretty_label = re.sub(r"_(\d+)", r"<sub>\1</sub>", label)
             point_labels += [pretty_label]
 
         fig.add_scatter3d(

diff --git a/pyproject.toml b/pyproject.toml
@@ -51,6 +51,7 @@ test = [
     "pytest-split>=0.9",
     "pytest>=8",
 ]
+brillouin = ["seekpath>=2.1"]
 
 [tool.setuptools.packages.find]
 include = ["pymatviz*"]