Testys commited on
Commit
8b5c2a1
·
verified ·
1 Parent(s): 70d8022

Update search_utils.py

Browse files
Files changed (1) hide show
  1. search_utils.py +1 -1
search_utils.py CHANGED
@@ -287,7 +287,7 @@ class SemanticSearch:
287
 
288
  # Deduplicate and sort results
289
  pre_dedup = len(results)
290
- results = results.drop_duplicates(subset=["title").sort_values("similarity", ascending=False).head(top_k)
291
  post_dedup = len(results)
292
 
293
  if pre_dedup > post_dedup:
 
287
 
288
  # Deduplicate and sort results
289
  pre_dedup = len(results)
290
+ results = results.drop_duplicates(subset=["title"]).sort_values("similarity", ascending=False).head(top_k)
291
  post_dedup = len(results)
292
 
293
  if pre_dedup > post_dedup: