Skip to content

Commit

Permalink
bug fix: cost time should be removed from the results of the serial_s…
Browse files Browse the repository at this point in the history
…earch

Signed-off-by: min.tian <[email protected]>
  • Loading branch information
alwayslove2013 authored and XuanYang-cn committed Dec 13, 2024
1 parent 50a59e9 commit d60166d
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions vectordb_bench/backend/task_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -206,15 +206,16 @@ def _load_train_data(self):
finally:
runner = None

def _serial_search(self) -> tuple[float, float]:
def _serial_search(self) -> tuple[float, float, float]:
"""Performance serial tests, search the entire test data once,
calculate the recall, serial_latency_p99
Returns:
tuple[float, float]: recall, serial_latency_p99
"""
try:
return self.serial_search_runner.run()
results, _ = self.serial_search_runner.run()
return results
except Exception as e:
log.warning(f"search error: {str(e)}, {e}")
self.stop()
Expand Down

0 comments on commit d60166d

Please sign in to comment.