Sort configs & splits alphabetically

#12
by sanchit-gandhi HF staff - opened
Files changed (1) hide show
  1. app.py +11 -10
app.py CHANGED
@@ -190,12 +190,12 @@ else:
190
  dataset_df = dataset_df.dropna(axis="columns", how="all")
191
 
192
  if len(dataset_df) > 0:
193
- config_set = set(dataset_df["config"])
 
 
194
  if "-unspecified-" in config_set:
195
- config_set.discard("-unspecified-")
196
- selectable_configs = ["-unspecified-"] + list(config_set)
197
- else:
198
- selectable_configs = list(config_set)
199
 
200
  if dataset != "-any-":
201
  config = st.sidebar.selectbox(
@@ -206,12 +206,13 @@ if len(dataset_df) > 0:
206
  )
207
  dataset_df = dataset_df[dataset_df.config == config]
208
 
209
- split_set = set(dataset_df["split"])
 
 
210
  if "-unspecified-" in split_set:
211
- split_set.discard("-unspecified-")
212
- selectable_splits = ["-unspecified-"] + list(split_set)
213
- else:
214
- selectable_splits = list(split_set)
215
  split = st.sidebar.selectbox(
216
  "Split",
217
  selectable_splits,
190
  dataset_df = dataset_df.dropna(axis="columns", how="all")
191
 
192
  if len(dataset_df) > 0:
193
+ selectable_configs = list(set(dataset_df["config"]))
194
+ selectable_configs.sort(key=lambda name: name.lower())
195
+
196
  if "-unspecified-" in config_set:
197
+ selectable_configs.discard("-unspecified-")
198
+ selectable_configs = ["-unspecified-"] + selectable_configs
 
 
199
 
200
  if dataset != "-any-":
201
  config = st.sidebar.selectbox(
206
  )
207
  dataset_df = dataset_df[dataset_df.config == config]
208
 
209
+ selectable_splits = list(set(dataset_df["split"]))
210
+ selectable_splits.sort(key=lambda name: name.lower())
211
+
212
  if "-unspecified-" in split_set:
213
+ selectable_splits.discard("-unspecified-")
214
+ selectable_splits = ["-unspecified-"] + selectable_splits
215
+
 
216
  split = st.sidebar.selectbox(
217
  "Split",
218
  selectable_splits,