From 88bfc263189bfbd9aec59e6b793fbb6111ccf09c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jordan=20Fr=C3=A9ry?= <jordan.frery@zama.ai>
Date: Wed, 13 Apr 2022 11:08:49 +0200
Subject: [PATCH] fix: update api in sanity_check

---
 docker/release_resources/sanity_check.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/release_resources/sanity_check.py b/docker/release_resources/sanity_check.py
index 8314e4f20..742ba2c8d 100644
--- 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