Include m_ll in process options

NNPDF · Nov 5, 2024 · 5761272 · 5761272
1 parent fd35640
commit 5761272
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/validphys2/src/validphys/process_options.py b/validphys2/src/validphys/process_options.py
@@ -36,6 +36,7 @@ class _Vars:
     abs_eta_2 = "abs_eta_2"
     eta_1 = "eta_1"
     eta_2 = "eta_2"
+    m_ll = "m_ll"
 
 
 class _KinematicsInformation:
@@ -249,7 +250,7 @@ def _dybosonpt_xq2map(kin_dict):
     Here pT refers to the transverse momentum of the boson.
     """
     pT = kin_dict[_Vars.pT]
-    m_Z2 = kin_dict[_Vars.m_Z2]
+    m_Z2 = kin_dict.get_one_of(_Vars.m_Z2, _Vars.m_W2, _Vars.m_ll)
     sqrts = kin_dict[_Vars.sqrts]
     ET2 = m_Z2 + pT * pT
     x = (np.sqrt(ET2) + pT) / sqrts
@@ -359,7 +360,7 @@ def _dybosonpt_xq2map(kin_dict):
 DY_PT = _Process(
     "DY_PT",
     "DY W or Z (2 leptons) + j boson transverse momentum",
-    accepted_variables=(_Vars.pT, _Vars.m_W2, _Vars.m_Z2, _Vars.sqrts),
+    accepted_variables=(_Vars.pT, _Vars.m_W2, _Vars.m_Z2, _Vars.sqrts, _Vars.y, _Vars.m_ll),
     xq2map_function=_dybosonpt_xq2map,
 )