def METHOD_NAME(dtype, nrows, column_info, n_parts, client):
    ncols, n_info = column_info
    X, y = make_regression(
        n_samples=nrows,
        n_features=ncols,
        n_informative=n_info,
        client=client,
        dtype=dtype,
    )
    lasso = Lasso(client=client)
    lasso.fit(X, y)
    y_hat = lasso.predict(X)
    assert r2_score(y.compute(), y_hat.compute()) >= 0.99