diff --git a/frontends/concrete-python/tests/conftest.py b/frontends/concrete-python/tests/conftest.py index 6bd0115113..dc58f6e644 100644 --- a/frontends/concrete-python/tests/conftest.py +++ b/frontends/concrete-python/tests/conftest.py @@ -11,6 +11,7 @@ import numpy as np import pytest +import warnings import tests from concrete import fhe @@ -302,6 +303,9 @@ def sanitize(values): if all(np.array_equal(e, a) for e, a in zip(expected, actual)): break + warnings.warn(UserWarning(f'Test fail ({i+1}/{retries}), regenerate keyset and retry')) + circuit.keygen(force=True, seed=i+1) + if i == retries - 1: message = f"""