fix bug

Signed-off-by: min.tian <[email protected]>
XuanYang-cn · Jan 9, 2025 · 6ee3dbe · 6ee3dbe
1 parent 032515f
commit 6ee3dbe
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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