def METHOD_NAME(client):
    X = DataFrame({"chars": ["a", "b"], "int": [0, 2]})
    X = dask_cudf.from_cudf(X, npartitions=2)
    cats = DataFrame({"chars": ["a", "b", "c"], "int": [0, 1, 2]})
    enc = OneHotEncoder(categories=cats, sparse=False)
    ref = cp.array(
        [[1.0, 0.0, 0.0, 1.0, 0.0, 0.0], [0.0, 1.0, 0.0, 0.0, 0.0, 1.0]]
    )
    res = enc.fit_transform(X)
    cp.testing.assert_array_equal(res.compute(), ref)