Skip to content

Commit

Permalink
addressing catagories_ type mismatch when auto by explicitly setting …
Browse files Browse the repository at this point in the history
…dtype on test data to object dask#964
  • Loading branch information
brian-methodical committed Feb 10, 2023
1 parent 91daec6 commit 3f4517d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/preprocessing/test_encoders.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@
from dask_ml._compat import DASK_2_20_0, PANDAS_VERSION
from dask_ml.utils import assert_estimator_equal

X = np.array([["a"], ["a"], ["b"], ["c"]])
X = np.array([["a"], ["a"], ["b"], ["c"]], dtype=object)
dX = da.from_array(X, 2)
df = pd.DataFrame(X, columns=["A"]).apply(lambda x: x.astype("category"))
ddf = dd.from_pandas(df, npartitions=2)
Expand Down

0 comments on commit 3f4517d

Please sign in to comment.