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