SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Raise exceptions instead of asserts (#13907)

Browse Source
This commit is contained in:
Максим Заякин
2021-10-07 12:14:23 +05:00
committed by GitHub
parent 5be59a3649
commit 0d309ce39a
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
utils/download_glue_data.py
View File

@@ -68,8 +68,10 @@ def format_mrpc(data_dir, path_to_data):
mrpc_test_file = os.path.join(mrpc_dir, "msr_paraphrase_test.txt") mrpc_test_file = os.path.join(mrpc_dir, "msr_paraphrase_test.txt")
urllib.request.urlretrieve(MRPC_TRAIN, mrpc_train_file) urllib.request.urlretrieve(MRPC_TRAIN, mrpc_train_file)
urllib.request.urlretrieve(MRPC_TEST, mrpc_test_file) urllib.request.urlretrieve(MRPC_TEST, mrpc_test_file)
assert os.path.isfile(mrpc_train_file), "Train data not found at %s" % mrpc_train_file if not os.path.isfile(mrpc_train_file):
assert os.path.isfile(mrpc_test_file), "Test data not found at %s" % mrpc_test_file raise ValueError(f"Train data not found at {mrpc_train_file}")
if not os.path.isfile(mrpc_test_file):
raise ValueError(f"Test data not found at {mrpc_test_file}")
urllib.request.urlretrieve(TASK2PATH["MRPC"], os.path.join(mrpc_dir, "dev_ids.tsv")) urllib.request.urlretrieve(TASK2PATH["MRPC"], os.path.join(mrpc_dir, "dev_ids.tsv"))
dev_ids = [] dev_ids = []
@@ -118,7 +120,8 @@ def get_tasks(task_names):
else: else:
tasks = [] tasks = []
for task_name in task_names: for task_name in task_names:
assert task_name in TASKS, "Task %s not found!" % task_name if task_name not in TASKS:
raise ValueError(f"Task {task_name} not found!")
tasks.append(task_name) tasks.append(task_name)
return tasks return tasks