More manual fixes in py files

Author: saganatt

machine_learning_hep/scripts-dhadrons/adjusting-run2-run3/add_pt_bins.py

@@ -54,8 +54,8 @@ def main():
         #                      hist.GetBinContent(hist.GetNbinsX()) * hist.GetBinWidth(hist.GetNbinsX())) /\
         #                     width_combined))
         # hist2.SetBinError(last_bin,
-        #                   math.sqrt((hist.GetBinError(hist.GetNbinsX() - 1) * hist.GetBinWidth(hist.GetNbinsX() - 1) /\
-        #                             width_combined) **2  +\
+        #                   math.sqrt((hist.GetBinError(hist.GetNbinsX() - 1) * hist.GetBinWidth(hist.GetNbinsX() - 1) \
+        #                             / width_combined) **2  +\
         #                             (hist.GetBinError(hist.GetNbinsX()) * hist.GetBinWidth(hist.GetNbinsX()) /\
         #                             width_combined) ** 2))
         # print(f"Setting bin {last_bin} low edge {hist2.GetBinLowEdge(last_bin)} " \

machine_learning_hep/scripts-dhadrons/adjusting-run2-run3/modify_crosssec_run2.py

@@ -52,14 +52,13 @@ def main():
                 bin1 = merge_bins[ind]
                 bin2 = merge_bins[ind] + 1
                 weight_sum = hist.GetBinWidth(bin1) + hist.GetBinWidth(bin2)
-                average = hist.GetBinContent(bin1) * hist.GetBinWidth(bin1) +\
-                          hist.GetBinContent(bin2) * hist.GetBinWidth(bin2)
                 hist2.SetBinContent(binn,
                         (hist.GetBinContent(bin1) * hist.GetBinWidth(bin1) +\
                          hist.GetBinContent(bin2) * hist.GetBinWidth(bin2)) /\
                         weight_sum)
-                hist2.SetBinError(binn, math.sqrt(((hist.GetBinWidth(bin1) * hist.GetBinError(bin1)) / weight_sum) ** 2. +\
-                                                  ((hist.GetBinWidth(bin2) * hist.GetBinError(bin2)) / weight_sum) ** 2.))
+                hist2.SetBinError(binn,
+                                  math.sqrt(((hist.GetBinWidth(bin1) * hist.GetBinError(bin1)) / weight_sum) ** 2. +\
+                                            ((hist.GetBinWidth(bin2) * hist.GetBinError(bin2)) / weight_sum) ** 2.))
                 ind += 1
             print(f"New bin {binn} low edge {hist2.GetBinLowEdge(binn)} "\
                   f"up edge {hist2.GetXaxis().GetBinUpEdge(binn)} "\

machine_learning_hep/scripts-dhadrons/adjusting-run2-run3/remove_high_pt.py

@@ -26,7 +26,6 @@ def main():
     args = parser.parse_args()
 
     with TFile(args.filename) as fin, TFile(args.outname, "recreate") as fout:
-        objnames = fin.GetListOfKeys()
         histnames = [key.GetName() for key in fin.GetListOfKeys() if args.histname in key.GetName()]
         for histname in histnames:
             hist = fin.Get(histname)

machine_learning_hep/scripts-dhadrons/debugging/check_parquet.py

@@ -5,7 +5,6 @@
 import pandas as pd
 
 """
-
 file: check_parquet.py
 brief: Examples of different checks on any parquet file produced by the MLHEP preprocessing steps.
 usage: python check_parquet.py AnalysisResultsReco_fPt1_2.parquet