style: pre-commit fixes

legend-exp · May 6, 2024 · 042217a · 042217a
1 parent 40eec51
commit 042217a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 11 deletions.
diff --git a/src/pygama/evt/modules/geds.py b/src/pygama/evt/modules/geds.py
@@ -109,7 +109,7 @@ def apply_xtalk_correction(
     )
     energies_corr = ak.from_regular(energies_corr)
     multiplicity_mask = ak.from_regular(multiplicity_mask)
-    tcm_id_array=ak.from_regular(tcm_id_array)
+    tcm_id_array = ak.from_regular(tcm_id_array)
 
     if mode == "energy":
         return types.VectorOfVectors(energies_corr[multiplicity_mask])
@@ -208,12 +208,10 @@ def apply_xtalk_correction_and_calibrate(
         xtalk_matrix_rawids,
     )
 
-
     calibrated_corr = ak.from_regular(calibrated_corr)
     multiplicity_mask = ak.from_regular(multiplicity_mask)
     tcm_id_array = ak.from_regular(tcm_id_array)
 
-
     if mode == "energy":
         return types.VectorOfVectors(calibrated_corr[multiplicity_mask])
     elif mode == "tcm_id":

diff --git a/src/pygama/evt/modules/xtalk.py b/src/pygama/evt/modules/xtalk.py
@@ -90,11 +90,9 @@ def build_energy_array(
         for name, file, group, column in tier_params:
             try:
                 # read the energy data
-                data = lh5.read(
-                        f"ch{channel}/{group}/{column}", file, idx=idx_events
-                    )
+                data = lh5.read(f"ch{channel}/{group}/{column}", file, idx=idx_events)
                 tbl.add_column(name, data)
-            except (lh5.exceptions.LH5DecodeError,KeyError):
+            except (lh5.exceptions.LH5DecodeError, KeyError):
                 tbl.add_column(name, types.Array(np.full_like(idx_events, np.nan)))
 
         res = tbl.eval(observable)
@@ -154,11 +152,10 @@ def filter_hits(
         for name, file, group, column in tier_params:
             try:
                 # read the energy data
-                data = lh5.read(
-                        f"ch{channel}/{group}/{column}", file, idx=idx_events)
-
+                data = lh5.read(f"ch{channel}/{group}/{column}", file, idx=idx_events)
+
                 tbl.add_column(name, data)
-            except (lh5.exceptions.LH5DecodeError,KeyError):
+            except (lh5.exceptions.LH5DecodeError, KeyError):
                 tbl.add_column(name, types.Array(np.full_like(idx_events, np.nan)))
 
         # add the corrected energy to the table