💬 Update type hints

arafune · Mar 18, 2024 · 2e8cf70 · 2e8cf70
1 parent 1a076af
commit 2e8cf70
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/arpes/plotting/utils.py b/src/arpes/plotting/utils.py
@@ -632,7 +632,7 @@ def lineplot_arr(
 
 
 def plot_arr(
-    arr: XrTypes,
+    arr: xr.DataArray,
     ax: Axes | None = None,
     over: AxesImage | None = None,
     mask: XrTypes | None = None,
@@ -663,7 +663,7 @@ def imshow_mask(
     mask: XrTypes,
     ax: Axes | None = None,
     over: AxesImage | None = None,
-    **kwargs: Incomplete,
+    **kwargs: Unpack[IMshowParam],
 ) -> None:
     """Plots a mask by using a fixed color and transparency."""
     assert over is not None
@@ -672,7 +672,7 @@ def imshow_mask(
         ax = plt.gca()
     assert isinstance(ax, Axes)
 
-    default_kwargs = {
+    default_kwargs: IMshowParam = {
         "origin": "lower",
         "aspect": ax.get_aspect(),
         "alpha": 1.0,
@@ -706,7 +706,7 @@ def imshow_arr(
     """Similar to plt.imshow but users different default origin, and sets appropriate extents.
 
     Args:
-        arr (XrTypes): ARPES data
+        arr (xr.DataArray): ARPES data
         ax (Axes): [TODO:description]
         over ([TODO:type]): [TODO:description]
         kwargs: pass to ax.imshow