Fixes nightly build 9/29 (#541)

* fix ci build * Update test_sklearn_tfidf_vectorizer_converter_pipeline.py
onnx · Sep 30, 2020 · 0736a35 · 0736a35
1 parent 7b1c5a8
commit 0736a35
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/tests/test_sklearn_nearest_neighbour_converter.py b/tests/test_sklearn_nearest_neighbour_converter.py
@@ -121,6 +121,9 @@ def test_model_knn_regressor(self):
     @unittest.skipIf(
         StrictVersion(onnxruntime.__version__) < StrictVersion("1.2.0"),
         reason="not available")
+    @unittest.skipIf(
+        StrictVersion(onnx.__version__) <= StrictVersion("1.6.0"),
+        reason="not available")
     def test_model_knn_regressor_radius(self):
         model, X = self._fit_model(RadiusNeighborsRegressor())
         model_onnx = convert_sklearn(model, "KNN regressor",

diff --git a/tests/test_sklearn_tfidf_vectorizer_converter_pipeline.py b/tests/test_sklearn_tfidf_vectorizer_converter_pipeline.py
@@ -20,8 +20,8 @@ class TestSklearnTfidfVectorizerPipeline(unittest.TestCase):
     def common_test_model_tfidf_vectorizer_pipeline_cls(
             self, kind=None, verbose=False):
         if kind == 'stop':
-            if ort_version.startswith('1.4'):
-                # regression with stopwords in onnxruntime 1.4
+            if ort_version.startswith('1.4') or ort_version.startswith('1.5'):
+                # regression with stopwords in onnxruntime 1.4, 1.5
                 stopwords = ['theh']
             else:
                 stopwords = ['the', 'and', 'is']