fix(frontend): fix decryption dtype

zama-ai · Oct 14, 2024 · 056f931 · 056f931
1 parent a9e03e2
commit 056f931
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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)
         )