diff --git a/src/ekobox/evol_pdf.py b/src/ekobox/evol_pdf.py index 15e837736..b629dd249 100644 --- 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 index 7d788277b..baf70726d 100644 --- 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