e-mon commited on
Commit
4598ba3
·
1 Parent(s): 54760ff
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -274,6 +274,8 @@ AVG_COLUMNS = ["AVG"] + list(TASK_AVG_NAME_MAP.keys())
274
 
275
 
276
  def plot_size_vs_score(df_filtered: pd.DataFrame) -> go.Figure:
 
 
277
  df = ORIGINAL_DF[ORIGINAL_DF[AutoEvalColumn.row_id.name].isin(df_filtered[AutoEvalColumn.row_id.name])]
278
  df = df[df["#Params (B)"] > 0]
279
  df = df[["model_name_for_query", "#Params (B)"] + AVG_COLUMNS]
@@ -325,6 +327,8 @@ def plot_size_vs_score(df_filtered: pd.DataFrame) -> go.Figure:
325
 
326
 
327
  def plot_average_scores(df_filtered: pd.DataFrame) -> go.Figure:
 
 
328
  df = ORIGINAL_DF[ORIGINAL_DF[AutoEvalColumn.row_id.name].isin(df_filtered[AutoEvalColumn.row_id.name])]
329
  df = df[["model_name_for_query"] + list(TASK_AVG_NAME_MAP.keys())]
330
  df = df.rename(columns={"model_name_for_query": "Model"})
 
274
 
275
 
276
  def plot_size_vs_score(df_filtered: pd.DataFrame) -> go.Figure:
277
+ if len(ORIGINAL_DF) == 0 or AutoEvalColumn.row_id.name not in ORIGINAL_DF.columns:
278
+ return go.Figure()
279
  df = ORIGINAL_DF[ORIGINAL_DF[AutoEvalColumn.row_id.name].isin(df_filtered[AutoEvalColumn.row_id.name])]
280
  df = df[df["#Params (B)"] > 0]
281
  df = df[["model_name_for_query", "#Params (B)"] + AVG_COLUMNS]
 
327
 
328
 
329
  def plot_average_scores(df_filtered: pd.DataFrame) -> go.Figure:
330
+ if len(ORIGINAL_DF) == 0 or AutoEvalColumn.row_id.name not in ORIGINAL_DF.columns:
331
+ return go.Figure()
332
  df = ORIGINAL_DF[ORIGINAL_DF[AutoEvalColumn.row_id.name].isin(df_filtered[AutoEvalColumn.row_id.name])]
333
  df = df[["model_name_for_query"] + list(TASK_AVG_NAME_MAP.keys())]
334
  df = df.rename(columns={"model_name_for_query": "Model"})