Skip to content

Commit 3b2c568

Browse files
committed
Add missing ptshape init
1 parent 17931a0 commit 3b2c568

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

machine_learning_hep/analysis/analyzerdhadrons.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,8 @@ def __init__(self, datap, case, typean, period):
149149

150150
self.root_objects = []
151151

152+
self.do_ptshape = datap.get("do_ptshape", False)
153+
152154
# Fitting
153155
self.p_performval = datap["analysis"].get("event_cand_validation", None)
154156

@@ -453,7 +455,7 @@ def do_eff(gen_hist, sel_hist, histname, outname, eff_case):
453455

454456
do_eff("h_gen_pr", "h_sel_pr", "eff", "Eff", "prompt")
455457
do_eff("h_gen_fd", "h_sel_fd", "eff_fd", "EffFD", "feed-down")
456-
if self.do_ptshape: # pylint: disable=no-member
458+
if self.do_ptshape:
457459
do_eff("h_gen_fd_ptshape", "h_sel_fd_ptshape", "eff_fd_ptshape", "EffFDPtShape", "feed-down")
458460

459461
@staticmethod

0 commit comments

Comments
 (0)