Fix default dataset. (#539)

stephenbach · web-flow · commit a3f95dda1985 · 2021-10-14T19:16:26.000-04:00
* Fix default dataset.

* Actually find the index rather than hardcode it.
diff --git a/promptsource/app.py b/promptsource/app.py
@@ -229,6 +229,7 @@ def get_infos(d_name):
         template_collection,
         state,
     )
+    ag_news_index = dataset_list.index("ag_news")
 
     #
     # Select a dataset - starts with ag_news
@@ -237,7 +238,7 @@ def get_infos(d_name):
         "Dataset",
         dataset_list,
         key="dataset_select",
-        index=4,  # AG_NEWS
+        index=ag_news_index,
         help="Select the dataset to work on.",
     )
 

Original file line number	Diff line number	Diff line change
`@@ -229,6 +229,7 @@ def get_infos(d_name):`
`229`	`229`	`template_collection,`
`230`	`230`	`state,`
`231`	`231`	`)`
	`232`	`+ ag_news_index = dataset_list.index("ag_news")`
`232`	`233`
`233`	`234`	`#`
`234`	`235`	`# Select a dataset - starts with ag_news`
`@@ -237,7 +238,7 @@ def get_infos(d_name):`
`237`	`238`	`"Dataset",`
`238`	`239`	`dataset_list,`
`239`	`240`	`key="dataset_select",`
`240`		`- index=4, # AG_NEWS`
	`241`	`+ index=ag_news_index,`
`241`	`242`	`help="Select the dataset to work on.",`
`242`	`243`	`)`
`243`	`244`