pre-commit fixes

Author: ggmarshall

scripts/blinding_calibration.py

@@ -7,6 +7,7 @@
 import argparse
 import json
 import logging
+import os
 import pickle as pkl
 
 os.environ["LGDO_CACHE"] = "false"

scripts/pars_hit_ecal.py

@@ -123,7 +123,7 @@ def baseline_tracking_plots(files, lh5_path, plot_options=None):
 def energy_cal_th(
     data: pd.Dataframe,
     energy_params: list[str],
-    cal_energy_params: list = None,
+    cal_energy_params: list | None = None,
     selection_string: str = "",
     hit_dict: dict | None = None,
     cut_parameters: dict[str, int] | None = None,
@@ -150,7 +150,7 @@ def energy_cal_th(
     results_dict = {}
     plot_dict = {}
     full_object_dict = {}
-    for energy_param, cal_energy_param in zip(energy_params,cal_energy_params):
+    for energy_param, cal_energy_param in zip(energy_params, cal_energy_params):
         full_object_dict[cal_energy_param] = calibrate_parameter(
             energy_param,
             selection_string,
@@ -164,9 +164,7 @@ def energy_cal_th(
             tail_weight=tail_weight,
         )
         full_object_dict[cal_energy_param].calibrate_parameter(data)
-        results_dict[cal_energy_param] = full_object_dict[
-            energy_param
-        ].get_results_dict(data)
+        results_dict[cal_energy_param] = full_object_dict[energy_param].get_results_dict(data)
         hit_dict.update(full_object_dict[cal_energy_param].hit_dict)
         if ~np.isnan(full_object_dict[cal_energy_param].pars).all():
             plot_dict[cal_energy_param] = (
@@ -211,11 +209,12 @@ def energy_cal_th(
     configs = LegendMetadata(path=args.configs)
     channel_dict = configs.on(args.timestamp, system=args.datatype)["snakemake_rules"]
     if args.tier == "hit":
-        channel_dict =channel_dict["pars_hit_ecal"]["inputs"]["ecal_config"][args.channel]
+        channel_dict = channel_dict["pars_hit_ecal"]["inputs"]["ecal_config"][args.channel]
     elif args.tier == "pht":
-        channel_dict =channel_dict["pars_pht_ecal"]["inputs"]["ecal_config"][args.channel]
+        channel_dict = channel_dict["pars_pht_ecal"]["inputs"]["ecal_config"][args.channel]
     else:
-        raise ValueError("invalid tier")
+        msg = "invalid tier"
+        raise ValueError(msg)
 
     kwarg_dict = Props.read_from(channel_dict)
 

scripts/pars_pht_aoecal.py

@@ -4,7 +4,6 @@
 import json
 import logging
 import os
-import re
 import pathlib
 import pickle as pkl
 from typing import Callable
@@ -197,7 +196,6 @@ def run_splitter(files):
 
 # run aoe cal
 if kwarg_dict.pop("run_aoe") is True:
-
     params = [
         kwarg_dict["final_cut_field"],
         kwarg_dict["current_param"],
@@ -243,7 +241,6 @@ def run_splitter(files):
     )
     data["is_pulser"] = mask[threshold_mask]
 
-
     pdf = eval(kwarg_dict.pop("pdf")) if "pdf" in kwarg_dict else standard_aoe
 
     mean_func = eval(kwarg_dict.pop("mean_func")) if "mean_func" in kwarg_dict else pol1

scripts/pars_pht_partcal.py

@@ -19,6 +19,7 @@
 
 log = logging.getLogger(__name__)
 
+
 def update_cal_dicts(cal_dicts, update_dict):
     if re.match(r"(\d{8})T(\d{6})Z", list(cal_dicts)[0]):
         for tstamp in cal_dicts:
@@ -30,9 +31,10 @@ def update_cal_dicts(cal_dicts, update_dict):
         cal_dicts.update(update_dict)
     return cal_dicts
 
+
 def partition_energy_cal_th(
     data: pd.Datframe,
-    hit_dicts:dict,
+    hit_dicts: dict,
     energy_params: list[str],
     selection_string: str = "",
     threshold: int = 0,
@@ -205,7 +207,7 @@ def run_splitter(files):
     params=params,
     threshold=kwarg_dict["threshold"],
     return_selection_mask=True,
-    cal_energy_param = kwarg_dict["energy_params"][0]
+    cal_energy_param=kwarg_dict["energy_params"][0],
 )
 
 # get pulser mask from tcm files
@@ -226,7 +228,10 @@ def run_splitter(files):
 
 # run energy supercal
 hit_dicts, ecal_results, plot_dict, ecal_obj = partition_energy_cal_th(
-    data, cal_dict, election_string=f"{kwarg_dict.pop('final_cut_field')}&(~is_pulser)", **kwarg_dict
+    data,
+    cal_dict,
+    election_string=f"{kwarg_dict.pop('final_cut_field')}&(~is_pulser)",
+    **kwarg_dict,
 )