From 6b93cb739d0293508f9118b7a0dd0e31ccb5b4b6 Mon Sep 17 00:00:00 2001 From: Stephen Bach Date: Tue, 12 Oct 2021 14:31:29 -0400 Subject: [PATCH 1/2] Fix default dataset. --- promptsource/app.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/promptsource/app.py b/promptsource/app.py index a6c60dfcb..333f4c52c 100644 --- a/promptsource/app.py +++ b/promptsource/app.py @@ -229,7 +229,7 @@ def get_infos(d_name): "Dataset", dataset_list, key="dataset_select", - index=12, # AG_NEWS + index=4, # AG_NEWS help="Select the dataset to work on.", ) From 57e2003c46dc608a7d4d06bf565ff5a517925415 Mon Sep 17 00:00:00 2001 From: Stephen Bach Date: Thu, 14 Oct 2021 19:11:47 -0400 Subject: [PATCH 2/2] Actually find the index rather than hardcode it. --- promptsource/app.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/promptsource/app.py b/promptsource/app.py index 6ad15d835..28b3d1153 100644 --- a/promptsource/app.py +++ b/promptsource/app.py @@ -227,6 +227,7 @@ def get_infos(d_name): template_collection, state, ) + ag_news_index = dataset_list.index("ag_news") # # Select a dataset - starts with ag_news @@ -235,7 +236,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.", )