From ed8a1782e801aea9d78699918de8095b67c4c9a3 Mon Sep 17 00:00:00 2001 From: Lukas Kreussel <65088241+LLukas22@users.noreply.github.com> Date: Fri, 18 Oct 2024 11:39:37 +0200 Subject: [PATCH] black --- tests/test_sklearn_ordinal_encoder.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/tests/test_sklearn_ordinal_encoder.py b/tests/test_sklearn_ordinal_encoder.py index f1555deea..80453774f 100644 --- a/tests/test_sklearn_ordinal_encoder.py +++ b/tests/test_sklearn_ordinal_encoder.py @@ -257,7 +257,7 @@ def test_model_ordinal_encoder_encoded_missing_value(self): ) assert_almost_equal(expected.reshape(-1), got[0].reshape(-1)) - + @unittest.skipIf( not ordinal_encoder_support(), reason="OrdinalEncoder was not available before 0.20", @@ -267,9 +267,9 @@ def test_model_ordinal_encoder_encoded_missing_value_no_nan(self): model = OrdinalEncoder(encoded_missing_value=42) data = np.array([["a"], ["b"], ["c"], ["d"]], dtype=np.object_) - + expected = model.fit_transform(data) - + model_onnx = convert_sklearn( model, "scikit-learn ordinal encoder", @@ -278,7 +278,10 @@ def test_model_ordinal_encoder_encoded_missing_value_no_nan(self): ) self.assertTrue(model_onnx is not None) dump_data_and_model( - data, model, model_onnx, basename="SklearnOrdinalEncoderEncodedMissingValueNoNan" + data, + model, + model_onnx, + basename="SklearnOrdinalEncoderEncodedMissingValueNoNan", ) sess = InferenceSession(