fix: update api in sanity_check

zama-ai · Apr 13, 2022 · 88bfc26 · 88bfc26
1 parent 0034200
commit 88bfc26
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/release_resources/sanity_check.py b/docker/release_resources/sanity_check.py
@@ -100,7 +100,7 @@ def function_to_compile(x):
     correct = 0
     for idx, (input_i, label_i) in enumerate(zip(inputs, labels), 1):
         print(f"Inference #{idx}")
-        result_i = engine.run(*input_i)
+        result_i = engine.encrypt_run_decrypt(*input_i)
 
         if result_i == label_i:
             correct += 1