Skip to content

Commit cd36f1e

Browse files
committed
added some debug info to process_chunk
1 parent 98256d8 commit cd36f1e

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

engine/base_client/search.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -266,10 +266,19 @@ def worker_function(self, distance, search_one, insert_one, chunk, result_queue,
266266

267267
def process_chunk(chunk, search_one, insert_one, insert_fraction):
268268
results = []
269+
insert_count = 0
270+
search_count = 0
271+
272+
#print(f"DEBUG: processing chunk of {len(chunk)} queries with insert_fraction={insert_fraction}")
273+
269274
for i, query in enumerate(chunk):
270275
if random.random() < insert_fraction:
271276
result = insert_one(query)
277+
insert_count += 1
272278
else:
273279
result = search_one(query)
280+
search_count += 1
274281
results.append(result)
282+
283+
#print(f"DEBUG: chunk complete. {search_count} searches, {insert_count} inserts")
275284
return results

0 commit comments

Comments
 (0)