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

Fix check for falsey inputs in run_summarization (#18155)

Browse Source
This commit is contained in:
John Giorgi
2022-07-18 00:50:32 -07:00
committed by GitHub
parent ccc0897804
commit c46d39f390
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/pytorch/summarization/run_summarization.py
View File

@@ -515,7 +515,7 @@ def main():
inputs, targets = [], [] inputs, targets = [], []
for i in range(len(examples[text_column])): for i in range(len(examples[text_column])):
if examples[text_column][i] is not None and examples[summary_column][i] is not None: if examples[text_column][i] and examples[summary_column][i]:
inputs.append(examples[text_column][i]) inputs.append(examples[text_column][i])
targets.append(examples[summary_column][i]) targets.append(examples[summary_column][i])