From c6f31e8618f10f2c8f9eddfdde497392dc3dc4bb Mon Sep 17 00:00:00 2001 From: Kelci Date: Wed, 11 Oct 2023 23:47:32 -0400 Subject: [PATCH] Fix missing not in VH overlap removal --- ewkcoffea/modules/selection_wwz.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ewkcoffea/modules/selection_wwz.py b/ewkcoffea/modules/selection_wwz.py index b58c6c47..175897c4 100644 --- a/ewkcoffea/modules/selection_wwz.py +++ b/ewkcoffea/modules/selection_wwz.py @@ -207,7 +207,7 @@ def add4lmask_wwz(events, year, isData, sample_name): genparts = events.GenPart is_zh = (abs(genparts[:,2].pdgId) == 23) # 3rd genparticle should be v for these samples is_w_from_h = ((abs(genparts.pdgId)==24) & (abs(genparts.distinctParent.pdgId) == 25)) - gen_mask = (is_zh & ak.any(is_w_from_h,axis=-1)) + gen_mask = ~(is_zh & ak.any(is_w_from_h,axis=-1)) mask = mask & gen_mask # TODO: Check if we need this, and add an if statement to not apply to data