💬 add type hints

arpes/endstations/plugin/ANTARES.py

@@ -75,7 +75,7 @@ def parse_axis_name_from_long_name(name, keep_segments=1, separator="_"):
     return separator.join(segments)
 
 
-def infer_scan_type_from_data(group):
+def infer_scan_type_from_data(group) -> str:
     """Determines the scan type for NeXuS format data.
 
     Because ANTARES stores every possible data type in the NeXuS file format, zeroing information

arpes/models/band.py

@@ -79,15 +79,15 @@ def fermi_velocity(self) -> xr.DataArray:
         return self.velocity.sel(eV=0, method="nearest")
 
     @property
-    def band_width(self):
+    def band_width(self) -> None:
         """The width along the band."""
         return
 
     def band_energy(self, coordinates):
         """The energy coordinate along the band."""
 
     @property
-    def self_energy(self):
+    def self_energy(self) -> None:
         """Calculates the self energy along the band."""
         return
 

arpes/utilities/ui.py

@@ -596,7 +596,7 @@ def translate_from_field(x):
 
         # close over the translation function
         def build_setter(translate, name):
-            def setter(value):
+            def setter(value) -> None:
                 try:
                     value = translate(value)
                 except ValueError:

arpes/xarray_extensions.py

@@ -2329,7 +2329,12 @@ def correct_angle_by(
 class GenericAccessorTools:
     _obj: xr.DataArray | xr.Dataset | None = None
 
-    def round_coordinates(self, coords, *, as_indices: bool = False) -> dict:
+    def round_coordinates(
+        self,
+        coords: dict[str, list[float] | NDArray[np.float_]],
+        *,
+        as_indices: bool = False,
+    ) -> dict:
         assert isinstance(self._obj, xr.DataArray | xr.Dataset)
         data = self._obj
         rounded = {