From 056f931331cfc429bf0a4a6c5b680233f0cf2572 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Alexandre=20P=C3=A9r=C3=A9?= Date: Mon, 14 Oct 2024 15:13:07 +0200 Subject: [PATCH] fix(frontend): fix decryption dtype --- .../concrete-python/concrete/fhe/compilation/client.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/frontends/concrete-python/concrete/fhe/compilation/client.py b/frontends/concrete-python/concrete/fhe/compilation/client.py index 02ae08c288..9b5b3fbc4d 100644 --- a/frontends/concrete-python/concrete/fhe/compilation/client.py +++ b/frontends/concrete-python/concrete/fhe/compilation/client.py @@ -294,7 +294,9 @@ def decrypt( decrypted = tuple( client_circuit.process_output( result._inner, position # pylint: disable=protected-access - ).to_py_val() + ) + .to_py_val() + .astype("int64") for position, result in enumerate(flattened_results) ) @@ -344,7 +346,9 @@ def simulate_decrypt( decrypted = tuple( client_circuit.simulate_process_output( result._inner, position # pylint: disable=protected-access - ).to_py_val() + ) + .to_py_val() + .astype("int64") for position, result in enumerate(flattened_results) )