nit

BlaiseMuhirwa · Feb 3, 2024 · 7337570 · 7337570
1 parent b1a4f87
commit 7337570
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/experiments/run-big-bench.py b/experiments/run-big-bench.py
@@ -137,15 +137,14 @@ def compute_metrics(
 
     """
     is_flatnav_index = type(index) in (flatnav.index.L2Index, flatnav.index.IPIndex)
+    index.set_num_threads(1)
     if is_flatnav_index:
-        index.set_num_threads(1)
         start = time.time()
         _, top_k_indices = index.search(
             queries=queries, ef_search=ef_search, K=k, num_initializations=300
         )
         end = time.time()
     else:
-        index.set_num_threads(1)
         index.set_ef(ef_search)
         start = time.time()
         # Search for HNSW return (ids, distances) instead of (distances, ids)