Fix check for falsey inputs in run_summarization (#18155)

2022-07-18 00:50:32 -07:00
parent ccc0897804
commit c46d39f390
1 changed files with 1 additions and 1 deletions
--- a/examples/pytorch/summarization/run_summarization.py
+++ b/examples/pytorch/summarization/run_summarization.py
@@ -515,7 +515,7 @@ def main():

        inputs, targets = [], []
        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])
                targets.append(examples[summary_column][i])