💬 update type annotation

arafune · Oct 17, 2023 · e768ad9 · e768ad9
1 parent cb2d999
commit e768ad9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/arpes/endstations/__init__.py b/arpes/endstations/__init__.py
@@ -199,7 +199,7 @@ def find_first_file(
         scan_desc: SCANDESC,
         *,
         allow_soft_match: bool = False,
-    ) -> NoReturn | None:
+    ) -> str | Path:
         """Attempts to find file associated to the scan given the user provided path or scan number.
 
         This is mostly done by regex matching over available options.
@@ -476,6 +476,7 @@ def load(self, scan_desc: SCANDESC | None = None, **kwargs: Incomplete) -> xr.Da
         self.trace("Resolving frame locations")
         resolved_frame_locations = self.resolve_frame_locations(scan_desc)
         resolved_frame_locations = [str(f) for f in resolved_frame_locations]
+        self.trace(f"resolved_frame_locations: {resolved_frame_locations}")
         if not resolved_frame_locations:
             msg = "File not found"
             raise RuntimeError(msg)

diff --git a/arpes/endstations/plugin/SPD_main.py b/arpes/endstations/plugin/SPD_main.py
@@ -115,14 +115,14 @@ def postprocess_final(
 
     def load_single_frame(
         self,
-        frame_path: str = "",
+        frame_path: str | Path = "",
         scan_desc: SCANDESC | None = None,
         **kwargs: str | float,
     ) -> xr.Dataset:
         """Load a single frame from an PHOIBOS 100 spectrometer with Prodigy.
 
         Args:
-            frame_path(str): _description_, by default ""
+            frame_path(str | Path): _description_, by default ""
             scan_desc(dict[str, str] | None): _description_, by default None
             kwargs(str | int | float): Pass to load_itx