add exception if requested output_distribution is normal

onnx · Feb 3, 2025 · 2017103 · 2017103
1 parent bf10071
commit 2017103
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/skl2onnx/operator_converters/quantile_transformer.py b/skl2onnx/operator_converters/quantile_transformer.py
@@ -39,6 +39,11 @@ def convert_quantile_transformer(scope: Scope,
     op_out = operator.outputs[0].full_name
     opv = container.target_opset
 
+    if op.output_distribution != 'uniform':
+         raise RuntimeError(
+             "Conversion of QuantileTransformer with output_distribution=%r "
+             "is not supported." % op.output_distribution)
+
     dtype = guess_numpy_type(op_in.type)
     if dtype != np.float64:
         dtype = np.float32