vectordb-bench 0.0.19__py3-none-any.whl → 0.0.20__py3-none-any.whl

vectordb_bench/frontend/components/custom/getCustomConfig.py

@@ -32,8 +32,7 @@ def get_custom_configs():
 
 def save_custom_configs(custom_configs: list[CustomDatasetConfig]):
     with open(config.CUSTOM_CONFIG_DIR, "w") as f:
-        json.dump([custom_config.dict()
-                   for custom_config in custom_configs], f, indent=4)
+        json.dump([custom_config.dict() for custom_config in custom_configs], f, indent=4)
 
 
 def generate_custom_case():

vectordb_bench/frontend/components/custom/initStyle.py

@@ -12,4 +12,4 @@ def initStyle(st):
             */
         </style>""",
         unsafe_allow_html=True,
-    )
+    )

vectordb_bench/frontend/components/get_results/saveAsImage.py

@@ -9,10 +9,12 @@ HTML_2_CANVAS_URL = "https://unpkg.com/html2canvas@1.4.1/dist/html2canvas.js"
 def load_unpkg(src: str) -> str:
     return requests.get(src).text
 
+
 def getResults(container, pageName="vectordb_bench"):
     container.subheader("Get results")
     saveAsImage(container, pageName)
 
+
 def saveAsImage(container, pageName):
     html2canvasJS = load_unpkg(HTML_2_CANVAS_URL)
     container.write()

vectordb_bench/frontend/components/run_test/caseSelector.py

@@ -1,6 +1,4 @@
-
 from vectordb_bench.frontend.config.styles import *
-from vectordb_bench.backend.cases import CaseType
 from vectordb_bench.frontend.config.dbCaseConfigs import *
 from collections import defaultdict
 
@@ -23,8 +21,7 @@ def caseSelector(st, activedDbList: list[DB]):
     dbToCaseConfigs = defaultdict(lambda: defaultdict(dict))
     caseClusters = UI_CASE_CLUSTERS + [get_custom_case_cluter()]
     for caseCluster in caseClusters:
-        activedCaseList += caseClusterExpander(
-            st, caseCluster, dbToCaseClusterConfigs, activedDbList)
+        activedCaseList += caseClusterExpander(st, caseCluster, dbToCaseClusterConfigs, activedDbList)
     for db in dbToCaseClusterConfigs:
         for uiCaseItem in dbToCaseClusterConfigs[db]:
             for case in uiCaseItem.cases:
@@ -40,8 +37,7 @@ def caseClusterExpander(st, caseCluster: UICaseItemCluster, dbToCaseClusterConfi
         if uiCaseItem.isLine:
             addHorizontalLine(expander)
         else:
-            activedCases += caseItemCheckbox(expander,
-                                             dbToCaseClusterConfigs, uiCaseItem, activedDbList)
+            activedCases += caseItemCheckbox(expander, dbToCaseClusterConfigs, uiCaseItem, activedDbList)
     return activedCases
 
 
@@ -53,9 +49,7 @@ def caseItemCheckbox(st, dbToCaseClusterConfigs, uiCaseItem: UICaseItem, actived
     )
 
     if selected:
-        caseConfigSetting(
-            st.container(), dbToCaseClusterConfigs, uiCaseItem, activedDbList
-        )
+        caseConfigSetting(st.container(), dbToCaseClusterConfigs, uiCaseItem, activedDbList)
 
     return uiCaseItem.cases if selected else []
 

vectordb_bench/frontend/components/run_test/dbConfigSetting.py

@@ -42,10 +42,7 @@ def dbConfigSettingItem(st, activeDb: DB):
 
     # db config (unique)
     for key, property in properties.items():
-        if (
-            key not in dbConfigClass.common_short_configs()
-            and key not in dbConfigClass.common_long_configs()
-        ):
+        if key not in dbConfigClass.common_short_configs() and key not in dbConfigClass.common_long_configs():
             column = columns[idx % DB_CONFIG_SETTING_COLUMNS]
             idx += 1
             dbConfig[key] = column.text_input(

vectordb_bench/frontend/components/run_test/dbSelector.py

@@ -22,7 +22,7 @@ def dbSelector(st):
         dbIsActived[db] = column.checkbox(db.name)
         try:
             column.image(DB_TO_ICON.get(db, ""))
-        except MediaFileStorageError as e:
+        except MediaFileStorageError:
             column.warning(f"{db.name} image not available")
             pass
     activedDbList = [db for db in DB_LIST if dbIsActived[db]]

vectordb_bench/frontend/components/run_test/generateTasks.py

@@ -7,13 +7,13 @@ def generate_tasks(activedDbList: list[DB], dbConfigs, activedCaseList: list[Cas
     for db in activedDbList:
         for case in activedCaseList:
             task = TaskConfig(
-                    db=db.value,
-                    db_config=dbConfigs[db],
-                    case_config=case,
-                    db_case_config=db.case_config_cls(
-                        allCaseConfigs[db][case].get(CaseConfigParamType.IndexType, None)
-                    )(**{key.value: value for key, value in allCaseConfigs[db][case].items()}),
-                )
+                db=db.value,
+                db_config=dbConfigs[db],
+                case_config=case,
+                db_case_config=db.case_config_cls(allCaseConfigs[db][case].get(CaseConfigParamType.IndexType, None))(
+                    **{key.value: value for key, value in allCaseConfigs[db][case].items()}
+                ),
+            )
             tasks.append(task)
-    
+
     return tasks

vectordb_bench/frontend/components/run_test/submitTask.py

@@ -1,6 +1,6 @@
 from datetime import datetime
-from vectordb_bench.frontend.config.styles import *
-from vectordb_bench.interface import benchMarkRunner
+from vectordb_bench.frontend.config import styles
+from vectordb_bench.interface import benchmark_runner
 
 
 def submitTask(st, tasks, isAllValid):
@@ -27,10 +27,8 @@ def submitTask(st, tasks, isAllValid):
 
 def taskLabelInput(st):
     defaultTaskLabel = datetime.now().strftime("%Y%m%d%H")
-    columns = st.columns(TASK_LABEL_INPUT_COLUMNS)
-    taskLabel = columns[0].text_input(
-        "task_label", defaultTaskLabel, label_visibility="collapsed"
-    )
+    columns = st.columns(styles.TASK_LABEL_INPUT_COLUMNS)
+    taskLabel = columns[0].text_input("task_label", defaultTaskLabel, label_visibility="collapsed")
     return taskLabel
 
 
@@ -46,10 +44,8 @@ def advancedSettings(st):
     )
 
     container = st.columns([1, 2])
-    k = container[0].number_input("k",min_value=1, value=100, label_visibility="collapsed")
-    container[1].caption(
-        "K value for number of nearest neighbors to search"
-    )
+    k = container[0].number_input("k", min_value=1, value=100, label_visibility="collapsed")
+    container[1].caption("K value for number of nearest neighbors to search")
 
     return index_already_exists, use_aliyun, k
 
@@ -58,20 +54,20 @@ def controlPanel(st, tasks, taskLabel, isAllValid):
     index_already_exists, use_aliyun, k = advancedSettings(st)
 
     def runHandler():
-        benchMarkRunner.set_drop_old(not index_already_exists)
+        benchmark_runner.set_drop_old(not index_already_exists)
         for task in tasks:
             task.case_config.k = k
-        benchMarkRunner.set_download_address(use_aliyun)
-        benchMarkRunner.run(tasks, taskLabel)
+        benchmark_runner.set_download_address(use_aliyun)
+        benchmark_runner.run(tasks, taskLabel)
 
     def stopHandler():
-        benchMarkRunner.stop_running()
+        benchmark_runner.stop_running()
 
-    isRunning = benchMarkRunner.has_running()
+    isRunning = benchmark_runner.has_running()
 
     if isRunning:
-        currentTaskId = benchMarkRunner.get_current_task_id()
-        tasksCount = benchMarkRunner.get_tasks_count()
+        currentTaskId = benchmark_runner.get_current_task_id()
+        tasksCount = benchmark_runner.get_tasks_count()
         text = f":running: Running Task {currentTaskId} / {tasksCount}"
         st.progress(currentTaskId / tasksCount, text=text)
 
@@ -89,7 +85,7 @@ def controlPanel(st, tasks, taskLabel, isAllValid):
         )
 
     else:
-        errorText = benchMarkRunner.latest_error or ""
+        errorText = benchmark_runner.latest_error or ""
         if len(errorText) > 0:
             st.error(errorText)
         disabled = True if len(tasks) == 0 or not isAllValid else False

vectordb_bench/frontend/components/tables/data.py

@@ -1,12 +1,11 @@
 from dataclasses import asdict
-from vectordb_bench.backend.cases import CaseType
-from vectordb_bench.interface import benchMarkRunner
+from vectordb_bench.interface import benchmark_runner
 from vectordb_bench.models import CaseResult, ResultLabel
 import pandas as pd
 
 
 def getNewResults():
-    allResults = benchMarkRunner.get_results()
+    allResults = benchmark_runner.get_results()
     newResults: list[CaseResult] = []
 
     for res in allResults:
@@ -14,7 +13,6 @@ def getNewResults():
         for result in results:
             if result.label == ResultLabel.NORMAL:
                 newResults.append(result)
-                    
 
     df = pd.DataFrame(formatData(newResults))
     return df
@@ -26,7 +24,6 @@ def formatData(caseResults: list[CaseResult]):
         db = caseResult.task_config.db.value
         db_label = caseResult.task_config.db_config.db_label
         case_config = caseResult.task_config.case_config
-        db_case_config = caseResult.task_config.db_case_config
         case = case_config.case_id.case_cls()
         filter_rate = case.filter_rate
         dataset = case.dataset.data.name
@@ -41,4 +38,4 @@ def formatData(caseResults: list[CaseResult]):
                 **metrics,
             }
         )
-    return data
+    return data

vectordb_bench/frontend/config/dbCaseConfigs.py

@@ -33,9 +33,9 @@ class UICaseItem(BaseModel):
     def __init__(
         self,
         isLine: bool = False,
-        case_id: CaseType = None,
-        custom_case: dict = {},
-        cases: list[CaseConfig] = [],
+        case_id: CaseType | None = None,
+        custom_case: dict | None = None,
+        cases: list[CaseConfig] | None = None,
         label: str = "",
         description: str = "",
         caseLabel: CaseLabel = CaseLabel.Performance,
@@ -70,17 +70,13 @@ class UICaseItemCluster(BaseModel):
 def get_custom_case_items() -> list[UICaseItem]:
     custom_configs = get_custom_configs()
     return [
-        UICaseItem(
-            case_id=CaseType.PerformanceCustomDataset, custom_case=custom_config.dict()
-        )
+        UICaseItem(case_id=CaseType.PerformanceCustomDataset, custom_case=custom_config.dict())
         for custom_config in custom_configs
     ]
 
 
 def get_custom_case_cluter() -> UICaseItemCluster:
-    return UICaseItemCluster(
-        label="Custom Search Performance Test", uiCaseItems=get_custom_case_items()
-    )
+    return UICaseItemCluster(label="Custom Search Performance Test", uiCaseItems=get_custom_case_items())
 
 
 UI_CASE_CLUSTERS: list[UICaseItemCluster] = [
@@ -224,8 +220,7 @@ CaseConfigParamInput_max_neighbors = CaseConfigInput(
         "max": 300,
         "value": 32,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.DISKANN.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.DISKANN.value,
 )
 
 CaseConfigParamInput_l_value_ib = CaseConfigInput(
@@ -236,8 +231,7 @@ CaseConfigParamInput_l_value_ib = CaseConfigInput(
         "max": 300,
         "value": 50,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.DISKANN.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.DISKANN.value,
 )
 
 CaseConfigParamInput_l_value_is = CaseConfigInput(
@@ -248,8 +242,7 @@ CaseConfigParamInput_l_value_is = CaseConfigInput(
         "max": 300,
         "value": 40,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.DISKANN.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.DISKANN.value,
 )
 
 CaseConfigParamInput_num_neighbors = CaseConfigInput(
@@ -260,8 +253,7 @@ CaseConfigParamInput_num_neighbors = CaseConfigInput(
         "max": 300,
         "value": 50,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.STREAMING_DISKANN.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.STREAMING_DISKANN.value,
 )
 
 CaseConfigParamInput_search_list_size = CaseConfigInput(
@@ -272,8 +264,7 @@ CaseConfigParamInput_search_list_size = CaseConfigInput(
         "max": 300,
         "value": 100,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.STREAMING_DISKANN.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.STREAMING_DISKANN.value,
 )
 
 CaseConfigParamInput_max_alpha = CaseConfigInput(
@@ -284,8 +275,7 @@ CaseConfigParamInput_max_alpha = CaseConfigInput(
         "max": 2.0,
         "value": 1.2,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.STREAMING_DISKANN.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.STREAMING_DISKANN.value,
 )
 
 CaseConfigParamInput_num_dimensions = CaseConfigInput(
@@ -296,8 +286,7 @@ CaseConfigParamInput_num_dimensions = CaseConfigInput(
         "max": 2000,
         "value": 0,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.STREAMING_DISKANN.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.STREAMING_DISKANN.value,
 )
 
 CaseConfigParamInput_query_search_list_size = CaseConfigInput(
@@ -308,8 +297,7 @@ CaseConfigParamInput_query_search_list_size = CaseConfigInput(
         "max": 150,
         "value": 100,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.STREAMING_DISKANN.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.STREAMING_DISKANN.value,
 )
 
 
@@ -321,8 +309,7 @@ CaseConfigParamInput_query_rescore = CaseConfigInput(
         "max": 150,
         "value": 50,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.STREAMING_DISKANN.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.STREAMING_DISKANN.value,
 )
 
 CaseConfigParamInput_IndexType_PgVector = CaseConfigInput(
@@ -358,8 +345,7 @@ CaseConfigParamInput_M = CaseConfigInput(
         "max": 64,
         "value": 30,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.HNSW.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.HNSW.value,
 )
 
 CaseConfigParamInput_m = CaseConfigInput(
@@ -370,8 +356,7 @@ CaseConfigParamInput_m = CaseConfigInput(
         "max": 64,
         "value": 16,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.HNSW.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.HNSW.value,
 )
 
 
@@ -383,8 +368,7 @@ CaseConfigParamInput_EFConstruction_Milvus = CaseConfigInput(
         "max": 512,
         "value": 360,
     },
-    isDisplayed=lambda config: config[CaseConfigParamType.IndexType]
-    == IndexType.HNSW.value,
+    isDisplayed=lambda config: config[CaseConfigParamType.IndexType] == IndexType.HNSW.value,
 )
 
 CaseConfigParamInput_EFConstruction_Weaviate = CaseConfigInput(
@@ -480,8 +464,7 @@ CaseConfigParamInput_EFConstruction_PgVectoRS = CaseConfigInput(
         "max": 2000,
         "value": 300,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.HNSW.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.HNSW.value,
 )
 
 CaseConfigParamInput_EFSearch_PgVectoRS = CaseConfigInput(
@@ -492,8 +475,7 @@ CaseConfigParamInput_EFSearch_PgVectoRS = CaseConfigInput(
         "max": 65535,
         "value": 100,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.HNSW.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.HNSW.value,
 )
 
 CaseConfigParamInput_EFConstruction_PgVector = CaseConfigInput(
@@ -504,8 +486,7 @@ CaseConfigParamInput_EFConstruction_PgVector = CaseConfigInput(
         "max": 1024,
         "value": 256,
     },
-    isDisplayed=lambda config: config[CaseConfigParamType.IndexType]
-    == IndexType.HNSW.value,
+    isDisplayed=lambda config: config[CaseConfigParamType.IndexType] == IndexType.HNSW.value,
 )
 
 
@@ -537,8 +518,7 @@ CaseConfigParamInput_EF_Milvus = CaseConfigInput(
         "max": MAX_STREAMLIT_INT,
         "value": 100,
     },
-    isDisplayed=lambda config: config[CaseConfigParamType.IndexType]
-    == IndexType.HNSW.value,
+    isDisplayed=lambda config: config[CaseConfigParamType.IndexType] == IndexType.HNSW.value,
 )
 
 CaseConfigParamInput_EF_Weaviate = CaseConfigInput(
@@ -565,8 +545,7 @@ CaseConfigParamInput_SearchList = CaseConfigInput(
         "max": MAX_STREAMLIT_INT,
         "value": 100,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.DISKANN.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.DISKANN.value,
 )
 
 CaseConfigParamInput_Nlist = CaseConfigInput(
@@ -611,8 +590,7 @@ CaseConfigParamInput_M_PQ = CaseConfigInput(
         "max": 65536,
         "value": 0,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    in [IndexType.GPU_IVF_PQ.value],
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) in [IndexType.GPU_IVF_PQ.value],
 )
 
 
@@ -624,8 +602,7 @@ CaseConfigParamInput_Nbits_PQ = CaseConfigInput(
         "max": 65536,
         "value": 8,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    in [IndexType.GPU_IVF_PQ.value],
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) in [IndexType.GPU_IVF_PQ.value],
 )
 
 CaseConfigParamInput_intermediate_graph_degree = CaseConfigInput(
@@ -636,8 +613,7 @@ CaseConfigParamInput_intermediate_graph_degree = CaseConfigInput(
         "max": 65536,
         "value": 64,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    in [IndexType.GPU_CAGRA.value],
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) in [IndexType.GPU_CAGRA.value],
 )
 
 CaseConfigParamInput_graph_degree = CaseConfigInput(
@@ -648,8 +624,7 @@ CaseConfigParamInput_graph_degree = CaseConfigInput(
         "max": 65536,
         "value": 32,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    in [IndexType.GPU_CAGRA.value],
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) in [IndexType.GPU_CAGRA.value],
 )
 
 CaseConfigParamInput_itopk_size = CaseConfigInput(
@@ -660,8 +635,7 @@ CaseConfigParamInput_itopk_size = CaseConfigInput(
         "max": 65536,
         "value": 128,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    in [IndexType.GPU_CAGRA.value],
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) in [IndexType.GPU_CAGRA.value],
 )
 
 CaseConfigParamInput_team_size = CaseConfigInput(
@@ -672,8 +646,7 @@ CaseConfigParamInput_team_size = CaseConfigInput(
         "max": 65536,
         "value": 0,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    in [IndexType.GPU_CAGRA.value],
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) in [IndexType.GPU_CAGRA.value],
 )
 
 CaseConfigParamInput_search_width = CaseConfigInput(
@@ -684,8 +657,7 @@ CaseConfigParamInput_search_width = CaseConfigInput(
         "max": 65536,
         "value": 4,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    in [IndexType.GPU_CAGRA.value],
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) in [IndexType.GPU_CAGRA.value],
 )
 
 CaseConfigParamInput_min_iterations = CaseConfigInput(
@@ -696,8 +668,7 @@ CaseConfigParamInput_min_iterations = CaseConfigInput(
         "max": 65536,
         "value": 0,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    in [IndexType.GPU_CAGRA.value],
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) in [IndexType.GPU_CAGRA.value],
 )
 
 CaseConfigParamInput_max_iterations = CaseConfigInput(
@@ -708,8 +679,7 @@ CaseConfigParamInput_max_iterations = CaseConfigInput(
         "max": 65536,
         "value": 0,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    in [IndexType.GPU_CAGRA.value],
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) in [IndexType.GPU_CAGRA.value],
 )
 
 CaseConfigParamInput_build_algo = CaseConfigInput(
@@ -718,8 +688,7 @@ CaseConfigParamInput_build_algo = CaseConfigInput(
     inputConfig={
         "options": ["IVF_PQ", "NN_DESCENT"],
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    in [IndexType.GPU_CAGRA.value],
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) in [IndexType.GPU_CAGRA.value],
 )
 
 
@@ -762,8 +731,7 @@ CaseConfigParamInput_Lists = CaseConfigInput(
         "max": 65536,
         "value": 10,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    in [IndexType.IVFFlat.value],
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) in [IndexType.IVFFlat.value],
 )
 
 CaseConfigParamInput_Probes = CaseConfigInput(
@@ -784,8 +752,7 @@ CaseConfigParamInput_Lists_PgVector = CaseConfigInput(
         "max": 65536,
         "value": 10,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.IVFFlat.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.IVFFlat.value,
 )
 
 CaseConfigParamInput_Probes_PgVector = CaseConfigInput(
@@ -796,8 +763,7 @@ CaseConfigParamInput_Probes_PgVector = CaseConfigInput(
         "max": 65536,
         "value": 1,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.IVFFlat.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.IVFFlat.value,
 )
 
 CaseConfigParamInput_EFSearch_PgVector = CaseConfigInput(
@@ -808,8 +774,7 @@ CaseConfigParamInput_EFSearch_PgVector = CaseConfigInput(
         "max": 2048,
         "value": 256,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None)
-    == IndexType.HNSW.value,
+    isDisplayed=lambda config: config.get(CaseConfigParamType.IndexType, None) == IndexType.HNSW.value,
 )
 
 
@@ -845,8 +810,7 @@ CaseConfigParamInput_QuantizationRatio_PgVectoRS = CaseConfigInput(
     inputConfig={
         "options": ["x4", "x8", "x16", "x32", "x64"],
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.quantizationType, None)
-    == "product"
+    isDisplayed=lambda config: config.get(CaseConfigParamType.quantizationType, None) == "product"
     and config.get(CaseConfigParamType.IndexType, None)
     in [
         IndexType.HNSW.value,
@@ -885,8 +849,7 @@ CaseConfigParamInput_reranking_PgVector = CaseConfigInput(
     inputConfig={
         "value": False,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.quantizationType, None)
-    == "bit"
+    isDisplayed=lambda config: config.get(CaseConfigParamType.quantizationType, None) == "bit",
 )
 
 CaseConfigParamInput_quantized_fetch_limit_PgVector = CaseConfigInput(
@@ -899,8 +862,8 @@ CaseConfigParamInput_quantized_fetch_limit_PgVector = CaseConfigInput(
         "max": 1000,
         "value": 200,
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.quantizationType, None)
-    == "bit" and config.get(CaseConfigParamType.reranking, False)
+    isDisplayed=lambda config: config.get(CaseConfigParamType.quantizationType, None) == "bit"
+    and config.get(CaseConfigParamType.reranking, False),
 )
 
 
@@ -908,12 +871,10 @@ CaseConfigParamInput_reranking_metric_PgVector = CaseConfigInput(
     label=CaseConfigParamType.rerankingMetric,
     inputType=InputType.Option,
     inputConfig={
-        "options": [
-            metric.value for metric in MetricType if metric.value not in ["HAMMING", "JACCARD"]
-        ],
+        "options": [metric.value for metric in MetricType if metric.value not in ["HAMMING", "JACCARD"]],
     },
-    isDisplayed=lambda config: config.get(CaseConfigParamType.quantizationType, None)
-    == "bit" and config.get(CaseConfigParamType.reranking, False)
+    isDisplayed=lambda config: config.get(CaseConfigParamType.quantizationType, None) == "bit"
+    and config.get(CaseConfigParamType.reranking, False),
 )
 
 
@@ -1131,7 +1092,10 @@ ESPerformanceConfig = [
     CaseConfigParamInput_NumCandidates_ES,
 ]
 
-AWSOpensearchLoadingConfig = [CaseConfigParamInput_EFConstruction_AWSOpensearch, CaseConfigParamInput_M_AWSOpensearch]
+AWSOpensearchLoadingConfig = [
+    CaseConfigParamInput_EFConstruction_AWSOpensearch,
+    CaseConfigParamInput_M_AWSOpensearch,
+]
 AWSOpenSearchPerformanceConfig = [
     CaseConfigParamInput_EFConstruction_AWSOpensearch,
     CaseConfigParamInput_M_AWSOpensearch,
@@ -1250,7 +1214,10 @@ AlloyDBPerformanceConfig = [
     CaseConfigParamInput_max_parallel_workers_AlloyDB,
 ]
 
-AliyunElasticsearchLoadingConfig = [CaseConfigParamInput_EFConstruction_AliES, CaseConfigParamInput_M_AliES]
+AliyunElasticsearchLoadingConfig = [
+    CaseConfigParamInput_EFConstruction_AliES,
+    CaseConfigParamInput_M_AliES,
+]
 AliyunElasticsearchPerformanceConfig = [
     CaseConfigParamInput_EFConstruction_AliES,
     CaseConfigParamInput_M_AliES,

vectordb_bench/frontend/pages/concurrent.py

@@ -9,7 +9,7 @@ from vectordb_bench.frontend.components.check_results.filters import getshownDat
 from vectordb_bench.frontend.components.concurrent.charts import drawChartsByCase
 from vectordb_bench.frontend.components.get_results.saveAsImage import getResults
 from vectordb_bench.frontend.config.styles import FAVICON
-from vectordb_bench.interface import benchMarkRunner
+from vectordb_bench.interface import benchmark_runner
 from vectordb_bench.models import TestResult
 
 
@@ -25,7 +25,7 @@ def main():
     # header
     drawHeaderIcon(st)
 
-    allResults = benchMarkRunner.get_results()
+    allResults = benchmark_runner.get_results()
 
     def check_conc_data(res: TestResult):
         case_results = res.results
@@ -57,9 +57,7 @@ def main():
 
     # main
     latency_type = st.radio("Latency Type", options=["latency_p99", "latency_avg"])
-    drawChartsByCase(
-        shownData, showCaseNames, st.container(), latency_type=latency_type
-    )
+    drawChartsByCase(shownData, showCaseNames, st.container(), latency_type=latency_type)
 
     # footer
     footer(st.container())