Update src/deepsparse/utils/onnx.py

neuralmagic · Jul 21, 2023 · 3ecbe5a · 3ecbe5a
1 parent 57c46d0
commit 3ecbe5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/deepsparse/utils/onnx.py b/src/deepsparse/utils/onnx.py
@@ -475,7 +475,7 @@ def truncate_onnx_embedding_model(
     return output_filepath, tmp_file
 
 
-def default_cached_outputs(model_path):
+def default_cached_outputs(model_path: str) -> List[bool]:
     """
     :param model_path: Path to a model
     :return A list of bools that indicates caching of all outputs except the first one.