diff --git a/vectordb_bench/frontend/config/dbCaseConfigs.py b/vectordb_bench/frontend/config/dbCaseConfigs.py index 429ff59d8..d46de5f89 100644 --- a/vectordb_bench/frontend/config/dbCaseConfigs.py +++ b/vectordb_bench/frontend/config/dbCaseConfigs.py @@ -950,6 +950,12 @@ class CaseConfigInput(BaseModel): CaseConfigParamInput_query_search_list_size, ] +PgDiskANNLoadConfig = [ + CaseConfigParamInput_IndexType_PgDiskANN, + CaseConfigParamInput_max_neighbors, + CaseConfigParamInput_l_value_ib, +] + PgDiskANNPerformanceConfig = [ CaseConfigParamInput_IndexType_PgDiskANN, CaseConfigParamInput_max_neighbors, @@ -986,6 +992,7 @@ class CaseConfigInput(BaseModel): CaseLabel.Performance: PgVectorScalePerformanceConfig, }, DB.PgDiskANN: { + CaseLabel.Load: PgDiskANNLoadConfig, CaseLabel.Performance: PgDiskANNPerformanceConfig, }, }