From aed3caacbf63e36557b2c9b59ca0a7820ae206b6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Francesc=20Mart=C3=AD=20Escofet?= Date: Thu, 4 Jul 2024 13:29:15 +0200 Subject: [PATCH] Add return type --- metalearners/_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metalearners/_utils.py b/metalearners/_utils.py index 97879900..bc450b62 100644 --- a/metalearners/_utils.py +++ b/metalearners/_utils.py @@ -509,7 +509,7 @@ def check_spox_installed() -> None: ) -def infer_dtype_and_shape_onnx(tensor): +def infer_dtype_and_shape_onnx(tensor) -> tuple[np.dtype, tuple]: check_onnx_installed() import onnx