yibum commited on
Commit
f4017ee
1 Parent(s): ad6d69e

update column filter

Browse files
Files changed (2) hide show
  1. app.py +12 -8
  2. src/display/utils.py +3 -1
app.py CHANGED
@@ -131,16 +131,20 @@ def filter_llm_provider_func(df: pd.DataFrame, llm_provider_query: list) -> pd.D
131
  return df[df["LLM Provider"].isin(llm_provider_query)]
132
 
133
 
134
- def filter_metric_area_func(df: pd.DataFrame, metric_area_query: list) -> pd.DataFrame:
135
- return df[df["Metric Area"].isin(metric_area_query)]
136
-
137
-
138
  def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
139
- always_here_cols = [
140
- AutoEvalColumn.model.name,
 
 
 
 
 
 
 
 
 
 
141
  ]
142
- # We use COLS to maintain sorting
143
- filtered_df = df[always_here_cols + [c for c in COLS if c in df.columns and c in columns]]
144
  return filtered_df
145
 
146
 
 
131
  return df[df["LLM Provider"].isin(llm_provider_query)]
132
 
133
 
 
 
 
 
134
  def select_columns(df: pd.DataFrame, columns: list) -> pd.DataFrame:
135
+ # always_here_cols = [
136
+ # AutoEvalColumn.model.name,
137
+ # ]
138
+ model_provider_col = [AutoEvalColumn.model_provider.name] if AutoEvalColumn.model_provider.name in columns else []
139
+ # We use COLS to maintain sortingx
140
+ filtered_df = df[
141
+ (
142
+ [AutoEvalColumn.model.name]
143
+ + model_provider_col
144
+ + [AutoEvalColumn.use_case_name.name]
145
+ + [c for c in COLS if c in df.columns and c in columns and c != AutoEvalColumn.model_provider.name]
146
+ )
147
  ]
 
 
148
  return filtered_df
149
 
150
 
src/display/utils.py CHANGED
@@ -26,7 +26,9 @@ auto_eval_column_dict.append(
26
  ["model", ColumnContent, ColumnContent("Model Name", "markdown", True, never_hidden=True)]
27
  )
28
  auto_eval_column_dict.append(["model_provider", ColumnContent, ColumnContent("LLM Provider", "markdown", True)])
29
- auto_eval_column_dict.append(["use_case_name", ColumnContent, ColumnContent("Use Case Name", "markdown", True)])
 
 
30
  auto_eval_column_dict.append(["use_case_type", ColumnContent, ColumnContent("Use Case Type", "markdown", False)])
31
  auto_eval_column_dict.append(
32
  ["accuracy_method", ColumnContent, ColumnContent("Accuracy Method", "markdown", False, never_hidden=True)]
 
26
  ["model", ColumnContent, ColumnContent("Model Name", "markdown", True, never_hidden=True)]
27
  )
28
  auto_eval_column_dict.append(["model_provider", ColumnContent, ColumnContent("LLM Provider", "markdown", True)])
29
+ auto_eval_column_dict.append(
30
+ ["use_case_name", ColumnContent, ColumnContent("Use Case Name", "markdown", True, never_hidden=True)]
31
+ )
32
  auto_eval_column_dict.append(["use_case_type", ColumnContent, ColumnContent("Use Case Type", "markdown", False)])
33
  auto_eval_column_dict.append(
34
  ["accuracy_method", ColumnContent, ColumnContent("Accuracy Method", "markdown", False, never_hidden=True)]