From 2ea487823033ff7fc35fb6caa1d57646f2e9107d Mon Sep 17 00:00:00 2001 From: Aron Jansen Date: Tue, 28 Nov 2023 13:11:29 +0100 Subject: [PATCH] Simplify input generation in msr test Co-authored-by: Juan M. Cruz-Martinez --- n3fit/src/n3fit/tests/test_msr.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/n3fit/src/n3fit/tests/test_msr.py b/n3fit/src/n3fit/tests/test_msr.py index 847ffc89f0..e51076741c 100644 --- a/n3fit/src/n3fit/tests/test_msr.py +++ b/n3fit/src/n3fit/tests/test_msr.py @@ -6,11 +6,9 @@ def apply_layer_to_fixed_input(layer): np.random.seed(422) - pdf_integrated = np.expand_dims(np.random.normal(size=(1, 14)), -1) - pdf_integrated = op.numpy_to_tensor(pdf_integrated) + pdf_integrated = op.numpy_to_tensor(np.random.normal(size=(1, 14,1))) - photon_integral = np.expand_dims(np.random.normal(size=(1,)), -1) - photon_integral = op.numpy_to_tensor(photon_integral) + photon_integral = op.numpy_to_tensor([np.random.normal(size=(1,1))]) return layer(pdf_integrated, photon_integral)