Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
BalduinLetzer committed Aug 8, 2024
1 parent 866617e commit ca0d0c4
Show file tree
Hide file tree
Showing 3 changed files with 13 additions and 13 deletions.
16 changes: 8 additions & 8 deletions hbw/trigger/categories.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,10 +51,10 @@ def catid_trigger_orth_mu(
) -> tuple[ak.Array, ak.Array]:
if results:
return events, (
results.steps.TrigEleMatch &
results.steps.SR_mu &
results.steps.ref_trigger_mu
)
results.steps.TrigEleMatch &
results.steps.SR_mu &
results.steps.ref_trigger_mu
)
else:
raise NotImplementedError("Category didn't receive a SelectionResult")

Expand All @@ -69,9 +69,9 @@ def catid_trigger_orth_ele(
) -> tuple[ak.Array, ak.Array]:
if results:
return events, (
results.steps.TrigMuMatch &
results.steps.SR_ele &
results.steps.ref_trigger_e
)
results.steps.TrigMuMatch &
results.steps.SR_ele &
results.steps.ref_trigger_e
)
else:
raise NotImplementedError("Category didn't receive a SelectionResult")
4 changes: 2 additions & 2 deletions hbw/trigger/plot_efficiencies.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,14 +29,14 @@

logger = law.logger.get_logger(__name__)

'''
"""
law run cf.PlotVariables1D --version v1 --config l22post \
--processes tt_dl --variables electron_pt-trig_bits --categories trig_mu \
--selector trigger_sel --producers event_weights,trigger_prod \
--plot-function hbw.trigger.plot_efficiencies.plot_efficiencies \
--variable-settings "electron_pt,rebin=5,x_min=0,x_max=100" \
--general-settings "bin_sel=Ele30_WPTight_Gsf;"
'''
"""


def plot_efficiencies(
Expand Down
6 changes: 3 additions & 3 deletions hbw/trigger/trigger_sel.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@


@selector(
uses=(four_vec({"Muon", "Electron"})) | {"TrigObj.*", },
uses=(four_vec({"Muon", "Electron"})) | {"TrigObj.*"},
exposed=True,
)
def trigger_sel(
Expand Down Expand Up @@ -56,11 +56,11 @@ def trigger_sel(
# set lepton objects (these will be the objects used after cf.ReduceColumns)
results.objects["Muon"]["Muon"] = masked_sorted_indices(
(results.aux["mu_mask_tight"] & (events.Muon.pt > 15)),
events.Muon.pt
events.Muon.pt,
)
results.objects["Electron"]["Electron"] = masked_sorted_indices(
(results.aux["e_mask_tight"] & (events.Electron.pt > 15)),
events.Electron.pt
events.Electron.pt,
)

# check for match between offline and HLT objects
Expand Down

0 comments on commit ca0d0c4

Please sign in to comment.