Skip to content

Commit

Permalink
addressing pylint errors
Browse files Browse the repository at this point in the history
  • Loading branch information
anpicci committed Dec 19, 2024
1 parent f71c9cb commit 9f240e0
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions analysis/topeft_run2/analysis_processor.py
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,7 @@ def __init__(self, samples, wc_names_lst=[], hist_lst=None, ecut_threshold=None,
self._samples = samples
self._wc_names_lst = wc_names_lst
self._dtype = dtype
self.3l_offZ_split = offZ_split
self.offZ_3l_split = offZ_split
self.tau_h_analysis = tau_h_analysis
self.fwd_analysis = fwd_analysis

Expand Down Expand Up @@ -592,7 +592,7 @@ def process(self, events):
select_cat_dict = json.load(ch_json_test)

# This dictionary keeps track of which selections go with which SR categories
if self.3l_offZ_split:
if self.offZ_3l_split:
import_sr_cat_dict = select_cat_dict["OFFZ_SPLIT_CH_LST_SR"]
elif self.tau_h_analysis:
import_sr_cat_dict = select_cat_dict["TAU_CH_LST_SR"]
Expand Down Expand Up @@ -657,7 +657,7 @@ def process(self, events):

# Get mask for events that have two sf os leps close to z peak
sfosz_3l_OnZ_mask = tc_es.get_Z_peak_mask(l_fo_conept_sorted_padded[:,0:3],pt_window=10.0)
if self.3l_offZ_split:
if self.offZ_3l_split:
sfosz_3l_OffZ_low_mask = tc_es.get_off_Z_mask_low(l_fo_conept_sorted_padded[:,0:3],pt_window=0.0)
sfosz_3l_OffZ_any_mask = tc_es.get_any_sfos_pair(l_fo_conept_sorted_padded[:,0:3])
else:
Expand Down Expand Up @@ -741,7 +741,7 @@ def process(self, events):
preselections.add("3l_m", (events.is3l & pass_trg & charge3l_m))
preselections.add("3l_onZ", (sfosz_3l_OnZ_mask))

if self.3l_offZ_split:
if self.offZ_3l_split:
preselections.add("3l_offZ_low", (sfosz_3l_OffZ_mask & sfosz_3l_OffZ_any_mask & sfosz_3l_OffZ_low_mask))
preselections.add("3l_offZ_high", (sfosz_3l_OffZ_mask & sfosz_3l_OffZ_any_mask & ~sfosz_3l_OffZ_low_mask))
preselections.add("3l_offZ_none", (sfosz_3l_OffZ_mask & ~sfosz_3l_OffZ_any_mask))
Expand Down Expand Up @@ -839,7 +839,7 @@ def process(self, events):
ptz = te_es.get_Z_pt(l_fo_conept_sorted_padded[:,0:3],10.0)
if self.tau_h_analysis:
ptz_wtau = (l0+tau0).pt
if self.3l_offZ_split:
if self.offZ_3l_split:
ptz = te_es.get_ll_pt(l_fo_conept_sorted_padded[:,0:3],10.0)
# Leading (b+l) pair pt
bjetsl = goodJets[isBtagJetsLoose][ak.argsort(goodJets[isBtagJetsLoose].pt, axis=-1, ascending=False)]
Expand Down Expand Up @@ -1074,7 +1074,7 @@ def process(self, events):
# Skip histos that are not defined (or not relevant) to given categories
if ((("j0" in dense_axis_name) and ("lj0pt" not in dense_axis_name)) & (("CRZ" in ch_name) or ("CRflip" in ch_name))): continue
if ((("j0" in dense_axis_name) and ("lj0pt" not in dense_axis_name)) & ("0j" in ch_name)): continue
if self.3l_offZ_split:
if self.offZ_3l_split:
if (("ptz" in dense_axis_name) & ("onZ" not in lep_chan) & ("offZ_high" not in lep_chan) & ("offZ_low" not in lep_chan)):continue
elif self.tau_h_analysis:
if (("ptz" in dense_axis_name) and ("onZ" not in lep_chan)): continue
Expand Down

0 comments on commit 9f240e0

Please sign in to comment.