Simplify input generation in msr test

Co-authored-by: Juan M. Cruz-Martinez <juacrumar@lairen.eu>
NNPDF · Nov 28, 2023 · 2ea4878 · 2ea4878
1 parent a28da06
commit 2ea4878
Showing 1 changed file with 2 additions and 4 deletions.
diff --git 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)