diff --git a/tests/protodash/test_PDASH.py b/tests/protodash/test_PDASH.py index 5d7e107..78e608f 100644 --- a/tests/protodash/test_PDASH.py +++ b/tests/protodash/test_PDASH.py @@ -33,7 +33,7 @@ def test_ProtoDashExplainer(self): original = original[:, 1:] # one hot encoding of all columns/features - onehot_encoder = OneHotEncoder(sparse=False) + onehot_encoder = OneHotEncoder(sparse_output=False) onehot_encoded = onehot_encoder.fit_transform(original) print(onehot_encoded.shape, original.shape)