fix bugs: display case_name and case_description

Signed-off-by: min.tian <[email protected]>
zilliztech · Jun 21, 2023 · 5c3d135 · 5c3d135
1 parent 59e67ca
commit 5c3d135
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/vectordb_bench/backend/cases.py b/vectordb_bench/backend/cases.py
@@ -44,6 +44,13 @@ def case_name(self) -> str:
         if c is not None:
             return c().name
         raise ValueError("Case unsupported")
+
+    @property
+    def case_description(self) -> str:
+        c = self.case_cls
+        if c is not None:
+            return c().description
+        raise ValueError("Case unsupported")
 
 
 class CaseLabel(Enum):

diff --git a/vectordb_bench/frontend/components/run_test/caseSelector.py b/vectordb_bench/frontend/components/run_test/caseSelector.py
@@ -33,9 +33,9 @@ def caseSelector(st, activedDbList):
 
 
 def caseItem(st, allCaseConfigs, case: CaseType, activedDbList):
-    selected = st.checkbox(case.casename)
+    selected = st.checkbox(case.case_name)
     st.markdown(
-        f"<div style='color: #1D2939; margin: -8px 0 20px {CHECKBOX_INDENT}px; font-size: 14px;'>{case.get()().description}</div>",
+        f"<div style='color: #1D2939; margin: -8px 0 20px {CHECKBOX_INDENT}px; font-size: 14px;'>{case.case_description}</div>",
         unsafe_allow_html=True,
     )