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

defensive programming + expand/correct README (#10295)

Browse Source
This commit is contained in:
Stas Bekman
2021-02-22 10:58:50 -08:00
committed by GitHub
parent 9e147d31f6
commit f991daed18
2 changed files with 166 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
examples/seq2seq/run_seq2seq.py
View File

@@ -403,10 +403,18 @@ def main():
text_column = dataset_columns[0] if dataset_columns is not None else column_names[0]
else:
text_column = data_args.text_column
if text_column not in column_names:
raise ValueError(
f"--text_column' value '{data_args.text_column}' needs to be one of: {', '.join(column_names)}"
)
if data_args.summary_column is None:
summary_column = dataset_columns[1] if dataset_columns is not None else column_names[1]
else:
summary_column = data_args.summary_column
if summary_column not in column_names:
raise ValueError(
f"--summary_column' value '{data_args.summary_column}' needs to be one of: {', '.join(column_names)}"
)
else:
# Get the language codes for input/target.
lang_search = re.match("translation_([a-z]+)_to_([a-z]+)", data_args.task)