Tristan Thrush commited on
Commit
4ab0962
1 Parent(s): 82c959c
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -158,6 +158,8 @@ st.experimental_set_query_params(**{"dataset": [dataset]})
158
  dataset_df = dataframe[dataframe.dataset == dataset]
159
  dataset_df = dataset_df.dropna(axis="columns", how="all")
160
 
 
 
161
  if len(dataset_df) > 0:
162
 
163
  selectable_configs = list(set(dataset_df["config"]))
@@ -174,8 +176,6 @@ if len(dataset_df) > 0:
174
  )
175
  dataset_df = dataset_df[dataset_df.split == split]
176
 
177
- selectable_metrics = list(filter(lambda column: column not in ("model_id", "dataset", "split", "config"), dataset_df.columns))
178
-
179
  dataset_df = dataset_df.filter(["model_id"] + selectable_metrics)
180
  dataset_df = dataset_df.dropna(thresh=2) # Want at least two non-na values (one for model_id and one for a metric).
181
 
 
158
  dataset_df = dataframe[dataframe.dataset == dataset]
159
  dataset_df = dataset_df.dropna(axis="columns", how="all")
160
 
161
+ selectable_metrics = list(filter(lambda column: column not in ("model_id", "dataset", "split", "config"), dataset_df.columns))
162
+
163
  if len(dataset_df) > 0:
164
 
165
  selectable_configs = list(set(dataset_df["config"]))
 
176
  )
177
  dataset_df = dataset_df[dataset_df.split == split]
178
 
 
 
179
  dataset_df = dataset_df.filter(["model_id"] + selectable_metrics)
180
  dataset_df = dataset_df.dropna(thresh=2) # Want at least two non-na values (one for model_id and one for a metric).
181