💬 update type hints

arafune · Nov 22, 2023 · 2e80291 · 2e80291
1 parent a11794e
commit 2e80291
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/arpes/utilities/widgets.py b/arpes/utilities/widgets.py
@@ -22,7 +22,7 @@
 
 if TYPE_CHECKING:
     from _typeshed import Incomplete
-    from PySide6.QtCore.Qt import CheckState
+    from PySide6.QtCore import CheckState
 
 __all__ = (
     "SubjectivePushButton",

diff --git a/arpes/utilities/xarray.py b/arpes/utilities/xarray.py
@@ -87,7 +87,7 @@ def lift_dataarray(  # unused
         g: Function operating on an xr.DataArray
     """
 
-    def g(arr: xr.DataArray, *args: Incomplete, **kwargs: Incomplete):
+    def g(arr: xr.DataArray, *args: Incomplete, **kwargs: Incomplete) -> xr.DataArray:
         return apply_dataarray(arr, f, *args, **kwargs)
 
     return g
@@ -101,7 +101,7 @@ def lift_dataarray_attrs(
     Produces a new xr.DataArray.
 
     Args:
-        f
+        f: Function to apply
 
     Returns:
         g: Function operating on the attributes of an xr.DataArray