Skip to content

Commit

Permalink
black
Browse files Browse the repository at this point in the history
  • Loading branch information
LLukas22 committed Oct 18, 2024
1 parent e0c8361 commit ed8a178
Showing 1 changed file with 7 additions and 4 deletions.
11 changes: 7 additions & 4 deletions tests/test_sklearn_ordinal_encoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand All @@ -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",
Expand All @@ -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(
Expand Down

0 comments on commit ed8a178

Please sign in to comment.