chore(frontend-python): Regenerate keyset on error

zama-ai · Jan 2, 2024 · 9a8b644 · 9a8b644
1 parent afb9420
commit 9a8b644
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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"""