diff --git a/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp b/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp
index 202bf906e..0bc4cace9 100644
--- a/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp
+++ b/pyg_lib/csrc/sampler/cpu/neighbor_kernel.cpp
@@ -224,10 +224,12 @@ class NeighborSampler {
     // Case 2: Multinomial sampling:
     else {
       at::Tensor index;
-      if (replace) { // at::multinomial has good perfomance only when replace=true, e.g. https://github.com/pytorch/pytorch/issues/11931#top
+      if (replace) {  // at::multinomial has good perfomance only when
+                      // replace=true, e.g.
+                      // https://github.com/pytorch/pytorch/issues/11931#top
         index = at::multinomial(weight, count, replace);
-      }
-      else { // An Efficient Algorithm for Biased Sampling: https://utopia.duth.gr/~pefraimi/research/data/2007EncOfAlg.pdf
+      } else {  // An Efficient Algorithm for Biased Sampling:
+                // https://utopia.duth.gr/~pefraimi/research/data/2007EncOfAlg.pdf
         const auto rand = at::empty_like(weight).uniform_();
         const auto key = (rand.log() / weight);
         index = std::get<1>(key.topk(count));