fix(frontend-python): test_matmul.test_constant_matmul, randomly fail…

…ing test
zama-ai · Aug 8, 2024 · 38600f8 · 38600f8
1 parent 46c115a
commit 38600f8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/frontends/concrete-python/tests/execution/test_matmul.py b/frontends/concrete-python/tests/execution/test_matmul.py
@@ -159,8 +159,8 @@ def rhs_function(x):
     lhs_function_circuit = lhs_function.compile(lhs_inputset, configuration)
     rhs_function_circuit = rhs_function.compile(rhs_inputset, configuration)
 
-    lhs_sample = np.random.randint(minimum, maximum, size=lhs_shape)
-    rhs_sample = np.random.randint(minimum, maximum, size=rhs_shape)
+    lhs_sample = lhs_inputset[-1]
+    rhs_sample = rhs_inputset[-1]
 
     helpers.check_execution(lhs_operator_circuit, lhs_operator, lhs_sample)
     helpers.check_execution(rhs_operator_circuit, rhs_operator, rhs_sample)