Rewrite weights

benchmarks/runners/benchmark_against_apfel.py

@@ -46,7 +46,7 @@ def run_external(
 
 def plain_assert_external(theory, obs, sf, yad):
     # APFEL has a discretization in Q2/m2
-    if sf == "FLcharm" and yad["Q2"] < 1.5*theory["mc"] ** 2:
+    if sf == "FLcharm" and yad["Q2"] < 1.5 * theory["mc"] ** 2:
         return dict(abs=3e-5)
     if sf == "FLbottom" and theory["mb"] ** 2 / 4 < yad["Q2"] < theory["mb"] ** 2:
         return dict(abs=5e-6)
@@ -106,13 +106,11 @@ def benchmark_NLO(self):
             assert_external=plain_assert_external,
         )
 
+
 def sv_assert_external(theory, obs, sf, yad):
     if np.isclose(theory["XIF"], 1) and np.isclose(theory["XIR"], 1):
         return plain_assert_external(theory, obs, sf, yad)
-    if (
-        sf == "FLbottom"
-        and theory["mb"] ** 2 / 4 < yad["Q2"] < theory["mb"] ** 2
-    ):
+    if sf == "FLbottom" and theory["mb"] ** 2 / 4 < yad["Q2"] < theory["mb"] ** 2:
         # APFEL has a discreization in Q2/m2
         return dict(abs=1e-5)
     if sf == "FLcharm" and yad["Q2"] < 7:
@@ -123,10 +121,7 @@ def sv_assert_external(theory, obs, sf, yad):
                 return dict(abs=2e-4)  # production threshold
             if sf[:2] == "F3" and yad["Q2"] > 900:
                 return dict(abs=3e-5)  # why does the thing go worse again?
-        if (
-            sf == "F3charm"
-            and yad["x"] < 2e-3
-        ):
+        if sf == "F3charm" and yad["x"] < 2e-3:
             # there is a cancelation between sbar and g going on:
             # each of the channels is O(1) with O(1e-3) accuracy
             return dict(abs=5e-3)
@@ -139,6 +134,7 @@ def sv_assert_external(theory, obs, sf, yad):
             return dict(abs=5e-2)
     return None
 
+
 @pytest.mark.commit_check
 class BenchmarkScaleVariations(ApfelBenchmark):
     """Vary factorization and renormalization scale"""
@@ -152,7 +148,10 @@ def benchmark_NLO(self):
         return self.run_external(
             1,
             ["CT14llo_NF3"],
-            {"XIR": None, "XIF": None,},
+            {
+                "XIR": None,
+                "XIF": None,
+            },
             {"prDIS": None},
             assert_external=sv_assert_external,
         )
@@ -209,7 +208,11 @@ def ffns_assert(theory, obs, sf, yad):
             if theory["NfFF"] < 5:
                 return plain_assert_external(theory, obs, sf, yad)
             # TODO https://github.com/N3PDF/yadism/wiki/2020_05_28-F2charm-FFNS4-low-Q2-non-zero
-            if theory["NfFF"] == 5 and sf[2:] in ["bottom", "total"] and yad["Q2"] < theory["mb"] ** 2:
+            if (
+                theory["NfFF"] == 5
+                and sf[2:] in ["bottom", "total"]
+                and yad["Q2"] < theory["mb"] ** 2
+            ):
                 return False
             return None
 
@@ -232,7 +235,10 @@ def benchmark_NLO_ZM_VFNS(self):
     @pytest.mark.fonll
     def benchmark_NLO_FONLL(self):
         def fonll_assert(theory, _obs, sf, yad):
-            if sf == "FLbottom" and theory["mb"] ** 2 / 4 < yad["Q2"] < theory["mb"] ** 2:
+            if (
+                sf == "FLbottom"
+                and theory["mb"] ** 2 / 4 < yad["Q2"] < theory["mb"] ** 2
+            ):
                 # APFEL has a discreization in Q2/m2
                 return dict(abs=5e-6)
             if sf == "FLcharm" and yad["Q2"] < 7 and theory["DAMP"] == 1:
@@ -243,7 +249,7 @@ def fonll_assert(theory, _obs, sf, yad):
             1,
             ["CT14llo_NF6"],
             {"FNS": self._db().theory_query.FNS == "FONLL-A", "DAMP": None},
-            assert_external=fonll_assert
+            assert_external=fonll_assert,
         )
 
 

benchmarks/runners/benchmark_against_qcdnum.py

@@ -27,7 +27,12 @@ def run_external(
                 PTO > 0
             ):  # by default we're running in FFNS3, so we can use the big runcard
                 obs_query &= self.db.obs_query.F2charm.exists()
-        return self.db.run_external(PTO, pdfs, theory_update, obs_query,)
+        return self.db.run_external(
+            PTO,
+            pdfs,
+            theory_update,
+            obs_query,
+        )
 
 
 @pytest.mark.quick_check

benchmarks/runners/pineappl.py

@@ -0,0 +1,50 @@
+# -*- coding: utf-8 -*-
+# pylint: skip-file
+import pathlib
+import numpy as np
+
+import yaml
+import lhapdf
+
+import yadism
+
+data = pathlib.Path(__file__).parent / "pineappl"
+
+# read input
+with open(data / "pineappl_sample_obs.yaml") as o:
+    obs = yaml.safe_load(o)
+with open(data / "pineappl_sample_theory.yaml") as o:
+    theory = yaml.safe_load(o)
+
+# run yadism
+out = yadism.run_yadism(theory, obs)
+out.dump_yaml_to_file(data / "pineappl_sample.yaml")
+
+# apply uonly pdf
+pdf_set = lhapdf.mkPDF("CT14llo_NF6", 0)
+pdf_out = out.apply_pdf(pdf_set)
+pdf_out.dump_yaml_to_file(data / "pineappl_sample_CT14llo_NF6_0000.yaml")
+
+# debug first element
+def list_pdfs(lhapdf_like, pids, Q2, xiF, xgrid):
+    # factorization scale
+    muF2 = Q2 * xiF ** 2
+    pdfs = np.zeros((len(pids), len(xgrid)))
+    for j, pid in enumerate(pids):
+        if not lhapdf_like.hasFlavor(pid):
+            continue
+        pdfs[j] = np.array([lhapdf_like.xfxQ2(pid, z, muF2) / z for z in xgrid])
+    return pdfs
+
+
+# first_esf = out["F2total"][0]
+# first_pdfs = list_pdfs(pdf_set,out["pids"],first_esf.Q2, out["xiF"], out["interpolation_xgrid"])
+# u_idx = out["pids"].index(1)
+# # store debug information
+# debug_first = {
+#     "x": out["interpolation_xgrid"].tolist(),
+#     "grid": first_esf.values[u_idx].tolist(),
+#     "xfx/x": first_pdfs[u_idx].tolist(),
+# }
+# with open(data / "debug_first.yaml", "w") as o:
+#     yaml.safe_dump(debug_first, o)

benchmarks/runners/pineappl/pineappl_default_grid.yaml

@@ -0,0 +1,50 @@
+- 1
+- 0.9309440808717544
+- 0.8627839323906108
+- 0.7956242522922756
+- 0.7295868442414312
+- 0.6648139482473823
+- 0.601472197967335
+- 0.5397572337880445
+- 0.4798989029610255
+- 0.4221667753589648
+- 0.3668753186482242
+- 0.31438740076927585
+- 0.2651137041582823
+- 0.2195041265003886
+- 0.17802566042569432
+- 0.14112080644440345
+- 0.10914375746330703
+- 0.08228122126204893
+- 0.060480028754447364
+- 0.04341491741702269
+- 0.030521584007828916
+- 0.02108918668378717
+- 0.014375068581090129
+- 0.009699159574043398
+- 0.006496206194633799
+- 0.004328500638820811
+- 0.0028738675812817515
+- 0.0019034634022867384
+- 0.0012586797144272762
+- 0.0008314068836488144
+- 0.0005487795323670796
+- 0.00036205449638139736
+- 0.00023878782918561914
+- 0.00015745605600841445
+- 0.00010381172986576898
+- 0.00006843744918967896
+- 0.00004511438394964044
+- 0.0000297384953722449
+- 0.000019602505002391748
+- 0.00001292101569074731
+- 0.000008516806677573355
+- 0.000005613757716930151
+- 0.0000037002272069854957
+- 0.000002438943292891682
+- 0.000001607585498470808
+- 0.0000010596094959101024
+- 0.0000006984208530700364
+- 0.00000046035014748963906
+- 0.0000003034304765867952
+- 0.00000019999999999999954