KNNImputer with distance as metric is not supported

[JonatanHauge]

Hi

I would like to convert my sklearn KNN imputer to an .onnx file, but the distance metric is not supported.
Can anybody please try to solve this?

Best regards,
Jonatan

[xadupre]

Can you tell more about the unsupported metric you tried to use?

[JonatanHauge]

When fitting a KNN imputer in sklearn you choose between either 'uniform' or 'distance' as the used metric to weigh the influence of the found neighbors. If I fit a KNN imputer with the choice 'distance', then I cannot convert the model to onnx.

The error I get is that the metric 'distance' is not supported.

Best regards, Jonatan

[xadupre]

Sorry for the delay, do you have a dummy example I could use to replicate the issue and fix it?