[BUG] clip not implemented for Dask-cuDF

[sarahyurick]

With Pandas and CPU Dask, I can do:

import pandas as pd
import dask.dataframe as dd

data = {
    "id": ["a", "b", "c", "d"],
    "score": [-0.5, 0.5, 4.5, 5.5]
}
pdf = pd.DataFrame(data)
ddf = dd.from_pandas(pdf, npartitions=2)

ddf["score"] = (
    ddf["score"].clip(lower=0, upper=5).round().astype(int)
)

which works as expected. However, with cuDF and Dask-cuDF:

import cudf
import dask_cudf

data = {
    "id": ["a", "b", "c", "d"],
    "score": [-0.5, 0.5, 4.5, 5.5]
}
cdf = cudf.DataFrame(data)
ddf = dask_cudf.from_cudf(cdf, npartitions=2)

ddf["score"] = (
    ddf["score"].clip(lower=0, upper=5).round().astype(int)
)

it fails with:

NotImplementedError: `axis is not yet supported in clip`

cc @VibhuJawa

[VibhuJawa]

CC: @rjzamora