From c46d39f390f5191646cffd56afe6a2b60037420f Mon Sep 17 00:00:00 2001 From: John Giorgi Date: Mon, 18 Jul 2022 00:50:32 -0700 Subject: [PATCH] Fix check for falsey inputs in run_summarization (#18155) --- examples/pytorch/summarization/run_summarization.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/pytorch/summarization/run_summarization.py b/examples/pytorch/summarization/run_summarization.py index 8d5007a503..0cbd73fe98 100755 --- 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])