refactor: remove the legacy directory

utils.py → src/utils.py

@@ -43,7 +43,7 @@ def search_table(df: pd.DataFrame, query: str) -> pd.DataFrame:
     return df[(df[COL_NAME_RETRIEVAL_MODEL].str.contains(query, case=False))]
 
 
-def get_default_cols(task: str, columns: list, add_fix_cols: bool = True) -> list:
+def get_default_cols(task: str, columns: list = [], add_fix_cols: bool = True) -> list:
     cols = []
     types = []
     if task == "qa":
@@ -59,7 +59,7 @@ def get_default_cols(task: str, columns: list, add_fix_cols: bool = True) -> lis
     for col_name, col_type in zip(cols_list, types_list):
         if col_name not in benchmark_list:
             continue
-        if col_name not in columns:
+        if columns and col_name not in columns:
             continue
         cols.append(col_name)
         types.append(col_type)

tests/test_utils.py

@@ -1,8 +1,7 @@
 import pandas as pd
 import pytest
 
-from utils import filter_models, search_table, filter_queries, select_columns, update_table_long_doc, get_iso_format_timestamp, get_default_cols
-from src.display.utils import COLS_QA
+from src.utils import filter_models, search_table, filter_queries, select_columns, update_table_long_doc, get_iso_format_timestamp, get_default_cols
 
 
 @pytest.fixture
@@ -90,6 +89,6 @@ def test_get_iso_format_timestamp():
 
 
 def test_get_default_cols():
-    cols, types = get_default_cols("qa", COLS_QA)
+    cols, types = get_default_cols("qa")
     for c, t in zip(cols, types):
         print(f"type({c}): {t}")