Fix evol_pdf

NNPDF · Oct 15, 2024 · e7d775c · e7d775c
1 parent b6efcd1
commit e7d775c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 8 deletions.
diff --git a/src/ekobox/evol_pdf.py b/src/ekobox/evol_pdf.py
@@ -78,11 +78,11 @@ def evolve_pdfs(
     with EKO.read(eko_path) as eko_output:
         for initial_PDF in initial_PDF_list:
             evolved_PDF_list.append(
-                apply.apply_pdf(eko_output, initial_PDF, targetgrid)
+                apply.apply_pdf(eko_output, initial_PDF, targetgrid)[0]
             )
 
         # separate by nf the evolgrid (and order per nf/q)
-        q2block_per_nf = regroup_evolgrid(eko_output.evolgrid)  # pylint: disable=E1101
+        q2block_per_nf = regroup_evolgrid(eko_output.evolgrid)
 
     # update info file
     if targetgrid is None:
@@ -128,7 +128,7 @@ def collect_blocks(evolved_PDF: dict, q2block_per_nf: dict, xgrid: list):
     # fake xfxQ2
     def pdf_xq2(pid, x, Q2):
         x_idx = xgrid.index(x)
-        return x * evolved_PDF[(Q2, nf)]["pdfs"][pid][x_idx]
+        return x * evolved_PDF[(Q2, nf)][pid][x_idx]
 
     # loop on nf patches
     for nf, q2grid in q2block_per_nf.items():

diff --git a/tests/ekobox/test_evol_pdf.py b/tests/ekobox/test_evol_pdf.py
@@ -113,11 +113,7 @@ def test_collect_blocks():
     def mk(eps):
         f = {}
         for ep in eps:
-            f[ep] = {
-                "pdfs": {
-                    pid: np.random.rand(len(xgrid)) for pid in br.flavor_basis_pids
-                }
-            }
+            f[ep] = {pid: np.random.rand(len(xgrid)) for pid in br.flavor_basis_pids}
         return f
 
     # basic