diff --git a/conftest.py b/conftest.py index 447dbaef1..fdbd14319 100644 --- 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 index 2d83d33cc..60a399b00 100644 --- 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 index a8e2c94c3..95bfe94b5 100644 --- 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 index 955427e76..e37e156ee 100644 --- 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