From ce22ea5ff1dc240ea100609d4c701c544bcc6fe5 Mon Sep 17 00:00:00 2001 From: jnke2016 Date: Tue, 7 Jan 2025 17:00:54 -0800 Subject: [PATCH] fix style --- cpp/src/community/leiden_impl.cuh | 15 ++++++++------- python/cugraph/cugraph/dask/community/leiden.py | 2 +- 2 files changed, 9 insertions(+), 8 deletions(-) diff --git a/cpp/src/community/leiden_impl.cuh b/cpp/src/community/leiden_impl.cuh index 558367b703..5a492d34a3 100644 --- a/cpp/src/community/leiden_impl.cuh +++ b/cpp/src/community/leiden_impl.cuh @@ -613,10 +613,12 @@ void relabel_cluster_ids(raft::handle_t const& handle, handle.get_comms(), unique_cluster_ids.size(), handle.get_stream()); std::vector cluster_ids_starts(cluster_ids_size_per_rank.size()); - std::exclusive_scan(cluster_ids_size_per_rank.begin(), cluster_ids_size_per_rank.end(), cluster_ids_starts.begin(), size_t{0}); - auto& comm = handle.get_comms(); - local_cluster_id_first = cluster_ids_starts[comm.get_rank()]; - + std::exclusive_scan(cluster_ids_size_per_rank.begin(), + cluster_ids_size_per_rank.end(), + cluster_ids_starts.begin(), + size_t{0}); + auto& comm = handle.get_comms(); + local_cluster_id_first = cluster_ids_starts[comm.get_rank()]; } rmm::device_uvector numbering_indices(unique_cluster_ids.size(), handle.get_stream()); @@ -714,11 +716,10 @@ std::pair leiden( clustering + local_num_verts, unique_cluster_ids.begin()); - detail::relabel_cluster_ids( - handle, unique_cluster_ids, clustering, local_num_verts); + handle, unique_cluster_ids, clustering, local_num_verts); return std::make_pair(dendrogram->num_levels(), modularity); } -} // namespace cugraph \ No newline at end of file +} // namespace cugraph diff --git a/python/cugraph/cugraph/dask/community/leiden.py b/python/cugraph/cugraph/dask/community/leiden.py index 1c327853c7..beff0b59b5 100644 --- a/python/cugraph/cugraph/dask/community/leiden.py +++ b/python/cugraph/cugraph/dask/community/leiden.py @@ -1,4 +1,4 @@ -# Copyright (c) 2022-2025, NVIDIA CORPORATION. +# Copyright (c) 2022-2024, NVIDIA CORPORATION. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License.