-
Notifications
You must be signed in to change notification settings - Fork 1
/
qcdiso_variations.py
81 lines (76 loc) · 2.84 KB
/
qcdiso_variations.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
from __future__ import annotations # needed for type annotations in > python 3.7
from typing import List
from code_generation.configuration import Configuration
from code_generation.systematics import SystematicShift
from .producers import electrons as electrons
from .producers import muons as muons
def add_qcdisoVariations(
configuration: Configuration, available_sample_types: List[str], era: str
):
#########################
# nonprompt lepton isolation
# up: Irel > 0.25
# down: Irel > 0.15
# nom: Irel > 0.2
#########################
configuration.add_shift(
SystematicShift(
name="QCDIsoMuUncUp",
shift_config={
"lep": {"mu_antiiso": 0.25},
},
producers={"lep": [muons.LooseMuons, muons.NumberOfLooseMuons, muons.TightMuons, muons.NumberOfTightMuons, muons.AntiTightMuons, muons.NumberOfAntiTightMuons]},
),
#samples=["data"]
samples=[
sample
for sample in available_sample_types
if sample not in ["embedding", "embedding_mc"]
],
)
configuration.add_shift(
SystematicShift(
name="QCDIsoElUncUp",
shift_config={
"lep": {"el_antiiso": 0.25},
},
producers={"lep": [electrons.LooseElectrons, electrons.NumberOfLooseElectrons, electrons.TightElectrons, electrons.NumberOfTightElectrons, electrons.AntiTightElectrons, electrons.NumberOfAntiTightElectrons]},
),
#samples=["data"]
samples=[
sample
for sample in available_sample_types
if sample not in ["embedding", "embedding_mc"]
],
)
configuration.add_shift(
SystematicShift(
name="QCDIsoMuUncDown",
shift_config={
"lep": {"mu_antiiso": 0.15, "loose_mu_iso": 0.15},
},
producers={"lep": [muons.LooseMuons, muons.NumberOfLooseMuons, muons.TightMuons, muons.NumberOfTightMuons, muons.AntiTightMuons, muons.NumberOfAntiTightMuons]},
),
#samples=["data"]
samples=[
sample
for sample in available_sample_types
if sample not in ["embedding", "embedding_mc"]
],
)
configuration.add_shift(
SystematicShift(
name="QCDIsoElUncDown",
shift_config={
"lep": {"el_antiiso": 0.15},
},
producers={"lep": [electrons.LooseElectrons, electrons.NumberOfLooseElectrons, electrons.TightElectrons, electrons.NumberOfTightElectrons, electrons.AntiTightElectrons, electrons.NumberOfAntiTightElectrons]},
),
#samples=["data"]
samples=[
sample
for sample in available_sample_types
if sample not in ["embedding", "embedding_mc"]
],
)
return configuration