def METHOD_NAME(input_type, client):
    # Testing that nothing crashes with 1D labels
    X, y = make_classification(n_samples=10000)
    if input_type == "array":
        dX = da.from_array(X)
        dy = da.from_array(y)
    elif input_type == "dataframe":
        X = cudf.DataFrame(X)
        y = cudf.Series(y)
        dX = dd.from_pandas(X, npartitions=1)
        dy = dd.from_pandas(y, npartitions=1)
    clf = dKNNClf()
    clf.fit(dX, dy)
    clf.predict(dX)