exc

xadupre · xadupre · commit 6a70ac59ba0c · 2024-08-30T14:16:06.000+02:00
diff --git a/onnx_array_api/_helpers.py b/onnx_array_api/_helpers.py
@@ -9,7 +9,7 @@ def np_dtype_to_tensor_dtype(dtype: Any):
     """
     try:
         dt = helper.np_dtype_to_tensor_dtype(dtype)
-    except KeyError:
+    except (KeyError, ValueError):
         if dtype == np.float32:
             dt = TensorProto.FLOAT
         elif dtype == np.float64:
