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/backend/clients/milvus/config.py b/vectordb_bench/backend/clients/milvus/config.py index 7d0df803..2a009581 100644 --- a/vectordb_bench/backend/clients/milvus/config.py +++ b/vectordb_bench/backend/clients/milvus/config.py @@ -4,7 +4,7 @@ class MilvusConfig(DBConfig): - uri: SecretStr = "http://localhost:19530" + uri: SecretStr = "http://10.102.5.190:19530" user: str | None = None password: SecretStr | None = None 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