From 6ee3dbeb98bba69fdad8c7b85a514a73b2a69493 Mon Sep 17 00:00:00 2001 From: "min.tian" Date: Thu, 9 Jan 2025 15:45:06 +0800 Subject: [PATCH] fix bug Signed-off-by: min.tian --- vectordb_bench/backend/assembler.py | 4 ++-- vectordb_bench/interface.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/vectordb_bench/backend/assembler.py b/vectordb_bench/backend/assembler.py index b81d315c..b28a7847 100644 --- a/vectordb_bench/backend/assembler.py +++ b/vectordb_bench/backend/assembler.py @@ -53,8 +53,8 @@ def assemble_all( _ = k.init_cls # sort by dataset size - for k, _ in db2runner: - db2runner[k].sort(key=lambda x: x.ca.dataset.data.size) + for _, runner in db2runner.items(): + runner.sort(key=lambda x: x.ca.dataset.data.size) all_runners = [] all_runners.extend(load_runners) diff --git a/vectordb_bench/interface.py b/vectordb_bench/interface.py index 615a9600..ebe12d2e 100644 --- a/vectordb_bench/interface.py +++ b/vectordb_bench/interface.py @@ -249,7 +249,7 @@ def _clear_running_task(self): def _run_async(self, conn: Connection) -> bool: log.info( - f"task submitted: id={self.running_task.run_id}, {self.running_task.task_label}, ", + f"task submitted: id={self.running_task.run_id}, {self.running_task.task_label}, " f"case number: {len(self.running_task.case_runners)}", ) global global_result_future