From bb16cffd10a6baa4450b765e13b66524ddd22e01 Mon Sep 17 00:00:00 2001 From: boccaff Date: Fri, 15 Nov 2024 00:40:52 +0000 Subject: [PATCH] fix: removed hardcoded optset and unused import Signed-off-by: boccaff --- skl2onnx/operator_converters/target_encoder.py | 1 - skl2onnx/shape_calculators/target_encoder.py | 1 - 2 files changed, 2 deletions(-) diff --git a/skl2onnx/operator_converters/target_encoder.py b/skl2onnx/operator_converters/target_encoder.py index c64fac6f8..858f758cf 100644 --- a/skl2onnx/operator_converters/target_encoder.py +++ b/skl2onnx/operator_converters/target_encoder.py @@ -75,7 +75,6 @@ def convert_sklearn_target_encoder( feature_column.onnx_name, label_encoder_output, op_domain="ai.onnx.ml", - op_version=2, **attrs, ) apply_reshape( diff --git a/skl2onnx/shape_calculators/target_encoder.py b/skl2onnx/shape_calculators/target_encoder.py index b559a240c..89ec73c72 100644 --- a/skl2onnx/shape_calculators/target_encoder.py +++ b/skl2onnx/shape_calculators/target_encoder.py @@ -1,7 +1,6 @@ # SPDX-License-Identifier: Apache-2.0 -import copy from ..common._registration import register_shape_calculator from ..common.data_types import FloatTensorType from ..common.data_types import Int64TensorType, StringTensorType