Skip to content
Merged
Prev Previous commit
Next Next commit
disable the right tests
  • Loading branch information
xadupre committed Nov 13, 2023
commit 3910302b6f5a7bce63fbdf5ff6d42fbcd3771217
4 changes: 1 addition & 3 deletions _unittests/ut_npx/test_sklearn_array_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from onnx.defs import onnx_opset_version
from sklearn import config_context, __version__ as sklearn_version
from sklearn.discriminant_analysis import LinearDiscriminantAnalysis
from onnx_array_api.ext_test_case import ExtTestCase, ignore_warnings, skipif_ci_windows
from onnx_array_api.ext_test_case import ExtTestCase, ignore_warnings
from onnx_array_api.npx.npx_numpy_tensors import EagerNumpyTensor


Expand All @@ -17,7 +17,6 @@ class TestSklearnArrayAPI(ExtTestCase):
reason="reshape ArrayAPI not followed",
)
@ignore_warnings(DeprecationWarning)
@skipif_ci_windows("Unstable on Windows.")
def test_sklearn_array_api_linear_discriminant(self):
X = np.array(
[[-1, -1], [-2, -1], [-3, -2], [1, 1], [2, 1], [3, 2]], dtype=np.float64
Expand All @@ -40,7 +39,6 @@ def test_sklearn_array_api_linear_discriminant(self):
reason="reshape ArrayAPI not followed",
)
@ignore_warnings(DeprecationWarning)
@skipif_ci_windows("Unstable on Windows.")
def test_sklearn_array_api_linear_discriminant_float32(self):
X = np.array(
[[-1, -1], [-2, -1], [-3, -2], [1, 1], [2, 1], [3, 2]], dtype=np.float32
Expand Down
8 changes: 5 additions & 3 deletions _unittests/ut_ort/test_sklearn_array_api_ort.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from onnx.defs import onnx_opset_version
from sklearn import config_context, __version__ as sklearn_version
from sklearn.discriminant_analysis import LinearDiscriminantAnalysis
from onnx_array_api.ext_test_case import ExtTestCase
from onnx_array_api.ext_test_case import ExtTestCase, skipif_ci_windows
from onnx_array_api.ort.ort_tensors import EagerOrtTensor, OrtTensor


Expand All @@ -16,7 +16,8 @@ class TestSklearnArrayAPIOrt(ExtTestCase):
Version(sklearn_version) <= Version("1.2.2"),
reason="reshape ArrayAPI not followed",
)
def test_sklearn_array_api_linear_discriminant(self):
@skipif_ci_windows("Unstable on Windows.")
def test_sklearn_array_api_linear_discriminant_ort(self):
X = np.array(
[[-1, -1], [-2, -1], [-3, -2], [1, 1], [2, 1], [3, 2]], dtype=np.float64
)
Expand All @@ -38,7 +39,8 @@ def test_sklearn_array_api_linear_discriminant(self):
Version(sklearn_version) <= Version("1.2.2"),
reason="reshape ArrayAPI not followed",
)
def test_sklearn_array_api_linear_discriminant_float32(self):
@skipif_ci_windows("Unstable on Windows.")
def test_sklearn_array_api_linear_discriminant_ort_float32(self):
X = np.array(
[[-1, -1], [-2, -1], [-3, -2], [1, 1], [2, 1], [3, 2]], dtype=np.float32
)
Expand Down