lp_hist none bug fix

rkansal47 · Jul 22, 2024 · 8b953f4 · 8b953f4
1 parent a99a8b9
commit 8b953f4
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/src/HHbbVV/processors/TTScaleFactorsSkimmer.py b/src/HHbbVV/processors/TTScaleFactorsSkimmer.py
@@ -493,7 +493,13 @@ def process(self, events: ak.Array):
             )
             self.dump_table(pddf, fname)
 
-        return {year: {dataset: {"totals": totals_dict, "cutflow": cutflow, "lp_hist": lp_hist}}}
+        ret_dict = {year: {dataset: {"totals": totals_dict, "cutflow": cutflow}}}
+
+        if lp_hist is not None:
+            ret_dict[year][dataset]["lp_hist"] = lp_hist
+
+        print(ret_dict)
+        return ret_dict
 
     def postprocess(self, accumulator):
         return accumulator

diff --git a/src/HHbbVV/processors/bbVVSkimmer.py b/src/HHbbVV/processors/bbVVSkimmer.py
@@ -796,7 +796,13 @@ def process(self, events: ak.Array):
         fname = events.behavior["__events_factory__"]._partition_key.replace("/", "_") + ".parquet"
         self.dump_table(pddf, fname)
 
-        return {year: {dataset: {"totals": totals_dict, "cutflow": cutflow, "lp_hist": lp_hist}}}
+        ret_dict = {year: {dataset: {"totals": totals_dict, "cutflow": cutflow}}}
+
+        if lp_hist is not None:
+            ret_dict[year][dataset]["lp_hist"] = lp_hist
+
+        print(ret_dict)
+        return ret_dict
 
     def postprocess(self, accumulator):
         return accumulator

diff --git a/src/condor/submit.py b/src/condor/submit.py
@@ -30,6 +30,7 @@ def get_site_vars(site):
         t2_local_prefix = Path("/ceph/cms/")
         t2_prefix = "root://redirector.t2.ucsd.edu:1095"
         if username == "rkansal":
+            # Reminder: need to re-copy this from /tmp whenever it expires (symlink?)
             proxy = "/home/users/rkansal/x509up_u31735"
         elif username == "annava":
             proxy = "/home/users/annava/projects/HHbbVV/test"