Skip to content
Merged
Prev Previous commit
Next Next commit
issue
  • Loading branch information
xadupre committed Nov 12, 2023
commit 2db8edb11bec8fa1fa135f04c2945069f5681b1c
3 changes: 0 additions & 3 deletions _unittests/ut_ort/test_ort_tensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@
from io import StringIO
import numpy as np
from onnx import TensorProto
from onnx.checker import check_model
from onnx.defs import onnx_opset_version
from onnx.reference import ReferenceEvaluator
from onnxruntime import InferenceSession
Expand Down Expand Up @@ -197,7 +196,6 @@ def impl(xa, xb):
def test_astype_w2(self):
f = absolute_inline(copy_inline(Input("A")).astype(DType(TensorProto.FLOAT)))
onx = f.to_onnx(constraints={"A": Float64[None]})
check_model(onx)
x = np.array([[-5, 6]], dtype=np.float64)
z = np.abs(x.astype(np.float32))
ref = InferenceSession(
Expand All @@ -209,7 +207,6 @@ def test_astype_w2(self):
def test_astype0_w2(self):
f = absolute_inline(copy_inline(Input("A")).astype(DType(TensorProto.FLOAT)))
onx = f.to_onnx(constraints={"A": Float64[None]})
check_model(onx)
x = np.array(-5, dtype=np.float64)
z = np.abs(x.astype(np.float32))
ref = InferenceSession(
Expand Down