fix dump polarized solution

NNPDF · Mar 7, 2024 · 1343b23 · 1343b23
1 parent 63f0cd0
commit 1343b23
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 78 deletions.
diff --git a/extras/data/machinery/generate/E142.py b/extras/data/machinery/generate/E142.py
@@ -1,42 +1,4 @@
-from .utils import load, obs_template
-
-
-def dump(src_path, target):
-    """Generate the input card for E142 measurements.
-
-    Parameters
-    ----------
-    src_path : str
-        input path
-
-    Returns
-    -------
-    dict
-        observables dictionary, corresponding to the runcard
-
-    """
-    obs = obs_template.copy()
-    data = load(src_path, 0, ["x", "Q2"])
-    dict_kins = [
-        dict(x=d["x"]["mid"], y=d["y"]["mid"], Q2=d["Q2"]["mid"]) for d in data
-    ]
-
-    obs["PolarizationDIS"] = 0.0 if "_F1" in target.parent.name else 1.0
-    observable_name = "F1_total" if "_F1" in target.parent.name else "g1_total"
-    obs["observables"] = {observable_name: dict_kins}
-    if "_ep_" in str(src_path.stem) or "_mup_" in str(src_path.stem):
-        obs["TargetDIS"] = "proton"
-    elif "_en_" in str(src_path.stem) or "_mun_" in str(src_path.stem):
-        obs["TargetDIS"] = "neutron"
-    elif "_ed_" in str(src_path.stem) or "_mud_" in str(src_path.stem):
-        obs["TargetDIS"] = "isoscalar"
-
-    # Details regarding the observables
-    obs["prDIS"] = "NC"
-    obs["ProjectileDIS"] = "electron"
-
-    return obs
-
+from .utils import dump_polarized as dump
 
 # renaming
 new_names = {

diff --git a/extras/data/machinery/generate/E143.py b/extras/data/machinery/generate/E143.py
@@ -1,42 +1,4 @@
-from .utils import load, obs_template
-
-
-def dump(src_path, target):
-    """Generate the input card for E143 measurements.
-
-    Parameters
-    ----------
-    src_path : str
-        input path
-
-    Returns
-    -------
-    dict
-        observables dictionary, corresponding to the runcard
-
-    """
-    obs = obs_template.copy()
-    data = load(src_path, 0, ["x", "Q2"])
-    dict_kins = [
-        dict(x=d["x"]["mid"], y=d["y"]["mid"], Q2=d["Q2"]["mid"]) for d in data
-    ]
-
-    obs["PolarizationDIS"] = 0.0 if "_F1" in target.parent.name else 1.0
-    observable_name = "F1_total" if "_F1" in target.parent.name else "g1_total"
-    obs["observables"] = {observable_name: dict_kins}
-    if "_ep_" in str(src_path.stem) or "_mup_" in str(src_path.stem):
-        obs["TargetDIS"] = "proton"
-    elif "_en_" in str(src_path.stem) or "_mun_" in str(src_path.stem):
-        obs["TargetDIS"] = "neutron"
-    elif "_ed_" in str(src_path.stem) or "_mud_" in str(src_path.stem):
-        obs["TargetDIS"] = "isoscalar"
-
-    # Details regarding the observables
-    obs["prDIS"] = "NC"
-    obs["ProjectileDIS"] = "electron"
-
-    return obs
-
+from .utils import dump_polarized as dump
 
 # renaming
 new_names = {