diff --git a/tests/test_sklearn_nearest_neighbour_converter.py b/tests/test_sklearn_nearest_neighbour_converter.py index b9a0ee8a1..2af29f9ac 100644 --- a/tests/test_sklearn_nearest_neighbour_converter.py +++ b/tests/test_sklearn_nearest_neighbour_converter.py @@ -430,7 +430,7 @@ def test_model_knn_classifier_binary_class(self): @unittest.skipIf( StrictVersion(onnxruntime.__version__) < StrictVersion("1.2.0"), reason="not available") - @unittest.skipIf(onnx_opset_version() < TARGET_OPSET, + @unittest.skipIf(onnx_opset_version() < 12, reason="needs higher target_opset") def test_model_knn_classifier_binary_class_radius(self): model, X = self._fit_model_binary_classification( @@ -470,6 +470,8 @@ def test_model_knn_classifier_multi_class(self): @unittest.skipIf( StrictVersion(onnxruntime.__version__) < StrictVersion("1.2.0"), reason="not available") + @unittest.skipIf(onnx_opset_version() < 12, + reason="needs higher target_opset") def test_model_knn_classifier_multi_class_radius(self): model, X = self._fit_model_multiclass_classification( RadiusNeighborsClassifier())