Add return type

Quantco · Jul 4, 2024 · aed3caa · aed3caa
1 parent d508843
commit aed3caa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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