diff --git a/cpp/include/raft/cluster/kmeans.cuh b/cpp/include/raft/cluster/kmeans.cuh index fa5db93a64..ee1fc83a9b 100644 --- a/cpp/include/raft/cluster/kmeans.cuh +++ b/cpp/include/raft/cluster/kmeans.cuh @@ -61,7 +61,7 @@ using KeyValueIndexOp = detail::KeyValueIndexOp; * params, * X, * std::nullopt, - * centroids, + * centroids.view(), * raft::make_scalar_view(&inertia), * raft::make_scalar_view(&n_iter)); * @endcode