fix: fix wrong flag

zama-ai · Jun 13, 2024 · cabc34b · cabc34b
1 parent d866668
commit cabc34b
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/conftest.py b/conftest.py
@@ -155,7 +155,7 @@ def default_configuration():
         fhe_simulation=False,
         fhe_execution=True,
         compress_input_ciphertexts=os.environ.get("USE_INPUT_COMPRESSION", "1") == "1",
-        compress_eval_keys=True,
+        compress_evaluation_keys=True,
     )
 
 

diff --git a/src/concrete/ml/quantization/quantized_module.py b/src/concrete/ml/quantization/quantized_module.py
@@ -889,7 +889,7 @@ def compile(
             fhe_simulation=False,
             fhe_execution=True,
             compress_input_ciphertexts=enable_input_compression,
-            compress_eval_keys=True,
+            compress_evaluation_keys=True,
         )
 
         self._is_compiled = True

diff --git a/src/concrete/ml/sklearn/base.py b/src/concrete/ml/sklearn/base.py
@@ -582,7 +582,7 @@ def compile(
             fhe_simulation=False,
             fhe_execution=True,
             compress_input_ciphertexts=enable_input_compression,
-            compress_eval_keys=True,
+            compress_evaluation_keys=True,
         )
 
         self._is_compiled = True

diff --git a/tests/torch/test_hybrid_converter.py b/tests/torch/test_hybrid_converter.py
@@ -41,7 +41,7 @@ def run_hybrid_llm_test(
     configuration = Configuration(
         single_precision=False,
         compress_input_ciphertexts=True,
-        compress_eval_keys=True,
+        compress_evaluation_keys=True,
     )
 
     # Create a hybrid model