fix comments

search.py

@@ -1,7 +1,9 @@
-import math, os, ujson, tqdm
+import os, ujson, tqdm
 import torch
 import torch.nn.functional as F
+
 from dotenv import load_dotenv
+
 from colbert import Searcher
 from colbert.search.index_storage import IndexScorer
 from colbert.search.strided_tensor import StridedTensor
@@ -89,18 +91,21 @@ def colbert_score_packed(Q, D_packed, D_lengths):
 
     scores = D_packed @ Q.T
 
-    return ColBERT.segmented_maxsim(scores, D_lengths)
+    # C++ : Calculate maxsim operation
+    scores = ColBERT.segmented_maxsim(scores, D_lengths)
+
+    return scores
 
 
 def score_pids(config, Q, pids, centroid_scores):
+    # C++ : Filter pids under the centroid score threshold
     idx = centroid_scores.max(-1).values >= config.centroid_score_threshold
-
     pids = IndexScorer.filter_pids(
         pids, centroid_scores, embeddings.codes, doclens,
         offsets, idx, config.ndocs
     )
 
-    # Rank final list of docs using full approximate embeddings (including residuals)
+    # C++ : Rank final list of docs using full approximate embeddings (including residuals)
     D_packed = IndexScorer.decompress_residuals(
         pids, doclens, offsets, bucket_weights, codec.reversed_bit_map,
         codec.decompression_lookup_table, embeddings.residuals, embeddings.codes,