401d78baaffb0751f2d9fb8f1c4c51a848d35020,tests/sklearn/test_ImputerConverter.py,TestSklearnImputerConverter,test_model_imputer,#TestSklearnImputerConverter#,13

Before Change

    def test_model_imputer(self):
        model = Imputer(missing_values="NaN", strategy="mean", axis=0)
        model.fit([[1, 2], [np.nan, 3], [7, 6]])
        model_onnx = convert_sklearn(model, "scikit-learn imputer", [Int64TensorType([1, 2])])
        self.assertTrue(model_onnx is not None)

    def test_imputer_int_inputs(self):

After Change

    def test_model_imputer(self):
        model = Imputer(missing_values="NaN", strategy="mean", axis=0)
        model.fit([[1, 2], [np.nan, 3], [7, 6]])
        model_onnx = convert_sklearn(model, "scikit-learn imputer", [("input", Int64TensorType([1, 2]))])
        self.assertTrue(model_onnx is not None)

    def test_imputer_int_inputs(self):

Instances