fix(frontend): notebook bug due to wrong dtype

zama-ai · Nov 14, 2024 · 30ce304 · 30ce304
1 parent 20d03d4
commit 30ce304
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/frontends/concrete-python/concrete/fhe/compilation/client.py b/frontends/concrete-python/concrete/fhe/compilation/client.py
@@ -194,7 +194,16 @@ def encrypt(
         client_circuit = client_program.get_client_circuit(function_name)
 
         exported = [
-            (None if arg is None else Value(client_circuit.prepare_input(Value_(arg), position)))
+            (
+                None
+                if arg is None
+                else Value(
+                    client_circuit.prepare_input(
+                        Value_(arg.astype(np.int64) if isinstance(arg, np.ndarray) else arg),
+                        position,
+                    )
+                )
+            )
             for position, arg in enumerate(ordered_sanitized_args)
         ]
 
@@ -240,7 +249,12 @@ def simulate_encrypt(
             (
                 None
                 if arg is None
-                else Value(client_circuit.simulate_prepare_input(Value_(arg), position))
+                else Value(
+                    client_circuit.simulate_prepare_input(
+                        Value_(arg.astype(np.int64) if isinstance(arg, np.ndarray) else arg),
+                        position,
+                    )
+                )
             )
             for position, arg in enumerate(ordered_sanitized_args)
         ]