diff --git a/Makefile b/Makefile index 5458b6ebf..a996d3c9c 100644 --- a/Makefile +++ b/Makefile @@ -788,7 +788,7 @@ check_utils_use_case: # This command does not use a make script because of obscure import issues with Skops on macOS .PHONY: update_encrypted_dataframe # Update encrypted data-frame's development files update_encrypted_dataframe: - poetry run python ./src/concrete/ml/pandas/_development.py + poetry run python -c "from concrete.ml.pandas._development import save_client_server; save_client_server()" .PHONY: check_symlinks # Check that no utils.py are found in use_case_examples check_symlinks: diff --git a/src/concrete/ml/pandas/_development.py b/src/concrete/ml/pandas/_development.py index 2858154c8..75384cd3d 100644 --- a/src/concrete/ml/pandas/_development.py +++ b/src/concrete/ml/pandas/_development.py @@ -12,7 +12,7 @@ from concrete import fhe -from ..quantization.quantized_module import _get_inputset_generator +from ..quantization.quantized_module import _get_inputset_generator script_dir = Path(__file__).parent @@ -115,13 +115,13 @@ def left_right_join_to_compile( @fhe.function({"value": "encrypted"}) def create_batch_2d(value): batch = fhe.zeros((DFApiV2StaticHelper.BATCH_SIZE, DFApiV2StaticHelper.N_DIMS_TRAINING)) - batch[0,0] = value + batch[0,0] = fhe.refresh(value) return batch @fhe.function({"value": "encrypted"}) def create_batch_1d(value): batch = fhe.zeros((DFApiV2StaticHelper.BATCH_SIZE, )) - batch[0] = value + batch[0] = fhe.refresh(value) return batch @fhe.function( @@ -138,7 +138,7 @@ def build_batch_2d( index1: Union[Tracer, int], index2: Union[Tracer, int], ): - batch[index1, index2] = value + batch[index1, index2] = fhe.refresh(value) return batch @fhe.function( @@ -153,7 +153,7 @@ def build_batch_1d( value: Union[Tracer, int], index1: Union[Tracer, int], ): - batch[index1] = value + batch[index1] = fhe.refresh(value) return batch composition = Wired(