Fix check for falsey inputs in run_summarization (#18155)

c46d39f3 · John Giorgi · GitHub · ccc08978 · c46d39f3
Unverified Commit c46d39f3 authored Jul 18, 2022 by John Giorgi Committed by GitHub Jul 18, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

examples/pytorch/summarization/run_summarization.py examples/pytorch/summarization/run_summarization.py +1 -1

No files found.
--- 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])