💬 Update type hints

arafune · Mar 14, 2024 · c6b2d9c · c6b2d9c
1 parent de57db4
commit c6b2d9c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/arpes/_typing.py b/src/arpes/_typing.py
@@ -79,7 +79,7 @@
 EMISSION_ANGLE = Literal["phi", "psi"]
 ANGLE = Literal["alpha", "beta", "chi", "theta"] | EMISSION_ANGLE
 
-LegendLocation = (
+LEGENDLOCATION = (
     Literal[
         "best",
         0,

diff --git a/src/arpes/plotting/stack_plot.py b/src/arpes/plotting/stack_plot.py
@@ -43,7 +43,7 @@
     from matplotlib.typing import ColorType, RGBAColorType, RGBColorType
     from numpy.typing import NDArray
 
-    from arpes._typing import ColorbarParam, LegendLocation, MPLPlotKwargsBasic
+    from arpes._typing import LEGENDLOCATION, ColorbarParam, MPLPlotKwargsBasic
 __all__ = (
     "stack_dispersion_plot",
     "flat_stack_plot",
@@ -77,7 +77,7 @@ def offset_scatter_plot(
     scale_coordinate: float = 0.5,
     ylim: tuple[float, float] | tuple[()] = (),
     fermi_level: float | None = None,
-    loc: LegendLocation = "upper left",
+    loc: LEGENDLOCATION = "upper left",
     figsize: tuple[float, float] = (11, 5),
     *,
     aux_errorbars: bool = True,
@@ -222,7 +222,7 @@ def flat_stack_plot(  # noqa: PLR0913
     figsize: tuple[float, float] = (7, 5),
     title: str = "",
     out: str | Path = "",
-    loc: LegendLocation = "upper left",
+    loc: LEGENDLOCATION = "upper left",
     **kwargs: Unpack[MPLPlotKwargsBasic],
 ) -> Path | tuple[Figure | None, Axes]:
     """Generates a stack plot with all the lines distinguished by color rather than offset.