fix issues

Author: xadupre

_doc/conf.py

@@ -108,7 +108,6 @@
     ("py:class", "default=sklearn.utils.metadata_routing.UNCHANGED"),
     ("py:class", "ModelProto"),
     ("py:class", "Module"),
-    ("py:class", "numpy._typing._nested_sequence._NestedSequence"),
     ("py:class", "torch.fx.passes.operator_support.OperatorSupport"),
     ("py:class", "torch.fx.proxy.TracerBase"),
     ("py:class", "torch.utils._pytree.Context"),
@@ -125,6 +124,7 @@
     ("py:func", ".*numpy[.].*"),
     ("py:func", ".*scipy[.].*"),
     # ("py:func", ".*torch.ops.higher_order.*"),
+    ("py:class", ".*numpy._typing.[.].*"),
     ("py:class", ".*onnxruntime[.].*"),
     ("py:meth", ".*onnxruntime[.].*"),
 ]

onnx_diagnostic/helpers.py

@@ -625,7 +625,7 @@ def from_array_ml_dtypes(arr: npt.ArrayLike, name: Optional[str] = None) -> Tens
     """
     import ml_dtypes
 
-    assert isinstance(arr, np.ndarray), f"arr must be of type np.ndarray, got {type(arr)}"
+    assert isinstance(arr, np.ndarray), f"arr must be of type numpy.ndarray, got {type(arr)}"
 
     tensor = TensorProto()
     tensor.dims.extend(arr.shape)

onnx_diagnostic/ort_session.py

@@ -129,7 +129,7 @@ def __init__(
 class InferenceSessionForNumpy(_InferenceSession):
     """
     Wraps an `onnxruntime.InferenceSession` to overload method `run`
-    to support :class:`np.ndarray`.
+    to support :class:`numpy.ndarray`.
 
     :param sess: model or inference session
     :param session_options: options