ENH Minor typing and formatiing fixes

slac-lcls · Mar 26, 2024 · ef85d77 · ef85d77
1 parent 743a878
commit ef85d77
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/lute/tasks/peakfinders.py b/lute/tasks/peakfinders.py
@@ -647,10 +647,13 @@ def _run(self) -> None:
                         unbondnbrs=False,
                     ).astype(numpy.uint16)
 
+                hdffh: Any
                 if self._task_parameters.mask_file is not None:
-                    mask *= numpy.load(self._task_parameters.mask_file).astype(
-                        numpy.uint16
-                    )
+                    with h5py.File(self._task_parameters.mask_file, "r") as hdffh:
+                        loaded_mask: NDArray[numpy.int] = hdffh["entry_1/data_1/mask"][
+                            :
+                        ]
+                        mask *= loaded_mask.astype(numpy.uint16)
 
                 file_writer: CxiWriter = CxiWriter(
                     outdir=self._task_parameters.outdir,