fix issue

Signed-off-by: xadupre <[email protected]>
onnx · Feb 27, 2025 · 0247853 · 0247853
1 parent df637d4
commit 0247853
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/test_sklearn_nearest_neighbour_converter.py b/tests/test_sklearn_nearest_neighbour_converter.py
@@ -1733,6 +1733,20 @@ def test_sklearn_knn_imputer_issue_2025(self):
             backend="onnxruntime",
         )
 
+    @unittest.skipIf(KNNImputer is None, reason="new in 0.22")
+    @unittest.skipIf(
+        pv.Version(ort_version) <= pv.Version("1.16.0"),
+        reason="onnxruntime not recent enough",
+    )
+    @ignore_warnings(category=DeprecationWarning)
+    @unittest.skipIf(
+        pv.Version(ort_version) < pv.Version("1.20.0"),
+        reason="onnxruntime not recent enough",
+    )
+    @unittest.skipIf(
+        sys.platform != "linux" and pv.Version(skl_version) < pv.Version("1.6.0"),
+        "investigate why topk returns different results",
+    )
     def test_knn_imputer_one_nan(self):
         import numpy as np
         import onnxruntime as rt