From e07a1155508d70c9babd262a365f8e559645aa1f Mon Sep 17 00:00:00 2001 From: Felix Hekhorn Date: Wed, 16 Oct 2024 10:09:57 +0300 Subject: [PATCH] Fix inv matching benchmark --- benchmarks/eko/benchmark_inverse_matching.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/benchmarks/eko/benchmark_inverse_matching.py b/benchmarks/eko/benchmark_inverse_matching.py index d63b776f0..1004a5481 100644 --- a/benchmarks/eko/benchmark_inverse_matching.py +++ b/benchmarks/eko/benchmark_inverse_matching.py @@ -90,14 +90,10 @@ def benchmark_inverse_matching(): with pytest.raises(AssertionError): np.testing.assert_allclose(op1_nf3.operator, op2_nf3.operator) - pdf1 = apply.apply_pdf(eko_output1, toy.mkPDF("ToyLH", 0)) - pdf2 = apply.apply_pdf(eko_output2, toy.mkPDF("ToyLH", 0)) + pdf1, _ = apply.apply_pdf(eko_output1, toy.mkPDF("ToyLH", 0)) + pdf2, _ = apply.apply_pdf(eko_output2, toy.mkPDF("ToyLH", 0)) # test that different PTO matching is applied correctly - np.testing.assert_allclose( - pdf1[(MC**2, 4)]["pdfs"][C_PID], pdf2[(MC**2, 4)]["pdfs"][C_PID] - ) + np.testing.assert_allclose(pdf1[(MC**2, 4)][C_PID], pdf2[(MC**2, 4)][C_PID]) with pytest.raises(AssertionError): - np.testing.assert_allclose( - pdf1[(MC**2, 3)]["pdfs"][C_PID], pdf2[(MC**2, 3)]["pdfs"][C_PID] - ) + np.testing.assert_allclose(pdf1[(MC**2, 3)][C_PID], pdf2[(MC**2, 3)][C_PID])