Metrics will separately record the duration of insert and optimize.

Signed-off-by: min.tian <[email protected]>
zilliztech · Jan 23, 2025 · 20c8461 · 20c8461
1 parent 78d6906
commit 20c8461
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/vectordb_bench/backend/task_runner.py b/vectordb_bench/backend/task_runner.py
@@ -154,6 +154,8 @@ def _run_perf_case(self, drop_old: bool = True) -> Metric:
                 if TaskStage.LOAD in self.config.stages:
                     _, load_dur = self._load_train_data()
                     build_dur = self._optimize()
+                    m.insert_duration = round(load_dur, 4)
+                    m.optimize_duration = round(build_dur, 4)
                     m.load_duration = round(load_dur + build_dur, 4)
                     log.info(
                         f"Finish loading the entire dataset into VectorDB,"