fix opset

Signed-off-by: xadupre <[email protected]>

Author: xadupre

skl2onnx/operator_converters/quantile_transformer.py

@@ -143,7 +143,8 @@ def create_interpolation(X_input, quantiles_init, references_init):
                 OnnxMul(current_index, np.array(0, dtype=np.int64), op_version=opv),
                 np.array([target_size // 2], dtype=np.int64),
                 op_version=opv,
-            )
+            ),
+            op_version=opv,
         )
         step.add_to(scope, container)
 

tests/test_sklearn_quantile_transformer.py

@@ -23,12 +23,11 @@ def test_quantile_transformer_simple(self):
             [("input", FloatTensorType([None, X.shape[1]]))],
             target_opset=TARGET_OPSET,
         )
-        self.assertTrue(model_onnx is not None)
         dump_data_and_model(
             X.astype(np.float32),
             model,
             model_onnx,
-            basename="SklearnQuantileTransformer",
+            basename="SklearnQuantileTransformerSimple",
         )
 
     def test_quantile_transformer_int(self):
@@ -40,12 +39,11 @@ def test_quantile_transformer_int(self):
             [("input", FloatTensorType([None, X.shape[1]]))],
             target_opset=TARGET_OPSET,
         )
-        self.assertTrue(model_onnx is not None)
         dump_data_and_model(
             X.astype(np.float32),
             model,
             model_onnx,
-            basename="SklearnQuantileTransformer",
+            basename="SklearnQuantileTransformerInt",
         )
 
     def test_quantile_transformer_nan(self):
@@ -60,12 +58,11 @@ def test_quantile_transformer_nan(self):
             [("input", FloatTensorType([None, X.shape[1]]))],
             target_opset=TARGET_OPSET,
         )
-        self.assertTrue(model_onnx is not None)
         dump_data_and_model(
             X.astype(np.float32),
             model,
             model_onnx,
-            basename="SklearnQuantileTransformer",
+            basename="SklearnQuantileTransformerNan",
         )