chore: fix cifar

zama-ai · Mar 28, 2024 · d9dbfee · d9dbfee
1 parent f213738
commit d9dbfee
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/use_case_examples/cifar/cifar_brevitas_training/evaluate_one_example_fhe.py b/use_case_examples/cifar/cifar_brevitas_training/evaluate_one_example_fhe.py
@@ -138,9 +138,8 @@ def wrapper(*args, **kwargs):
     print(f"Quantization of a single input (image) took {quantization_execution_time} seconds")
     print(f"Size of CLEAR input is {q_x_numpy.nbytes} bytes\n")
 
-    p_error = quantized_numpy_module.fhe_circuit.p_error
     expected_quantized_prediction, clear_inference_time = measure_execution_time(
-        partial(quantized_numpy_module.fhe_circuit.simulate, p_error=p_error)
+        partial(quantized_numpy_module.fhe_circuit.simulate)
     )(q_x_numpy)
 
     # Encrypt the input