github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-04-17 03:00:54 -04:00
Code Issues Packages Projects Releases Wiki Activity

test(frontend-python): increase the number of retries for encrypted matmul and encrypted dot tests

Browse Source
This commit is contained in:
Umut
2023-05-17 13:12:43 +02:00
parent b174f81687
commit 01850f68e6
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
frontends/concrete-python/tests/execution/test_dot.py
View File

@@ -87,4 +87,4 @@ def test_dot_enc_enc(size, bitwidth, negative_only, signed, helpers):
np.random.randint(minv, maxv, size=(size,)),
]
helpers.check_execution(dot_enc_enc_function_circuit, dot_enc_enc_function, sample)
helpers.check_execution(dot_enc_enc_function_circuit, dot_enc_enc_function, sample, retries=3)

4
frontends/concrete-python/tests/execution/test_matmul.py
View File

@@ -328,7 +328,7 @@ def test_matmul_enc_enc_and_clear(lhs_shape, rhs_shape, bounds, helpers):
minimum, maximum, size=lhs_shape
), np.random.randint(minimum, maximum, size=rhs_shape)
helpers.check_execution(dual_op_circuit, func, [lhs_sample, rhs_sample])
helpers.check_execution(dual_op_circuit, func, [lhs_sample, rhs_sample], retries=3)
@pytest.mark.parametrize("bitwidth", [4, 10])
@@ -362,4 +362,4 @@ def test_matmul_zero(bitwidth, signed, helpers):
range_lhs[0], range_lhs[1], size=lhs_shape
), np.zeros(rhs_shape, dtype=np.int64)
helpers.check_execution(dual_op_circuit, enc_function_xy, [lhs_sample, rhs_sample])
helpers.check_execution(dual_op_circuit, enc_function_xy, [lhs_sample, rhs_sample], retries=3)