From 3d962f0b5c7f700f6e6d11f9421305c2e82de7bb Mon Sep 17 00:00:00 2001 From: Adam Kariv Date: Wed, 17 Jul 2024 14:19:57 +0300 Subject: [PATCH] Fix filter_ready_to_publish --- operators/manual_data_entry/external.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/operators/manual_data_entry/external.py b/operators/manual_data_entry/external.py index 0e22280..0ce3dcd 100644 --- a/operators/manual_data_entry/external.py +++ b/operators/manual_data_entry/external.py @@ -12,11 +12,12 @@ CHECKPOINT = 'external-mde' def filter_ready_to_publish(stats: Stats): - def func(row): - if row.get('Status') == 'בייצור': - return True - stats.increase('External Manual Data: Entry not ready to publish') - return False + def func(rows): + for row in rows: + if row.get('Status') == 'בייצור': + yield row + else: + stats.increase('External Manual Data: Entry not ready to publish') return func def fetch_google_spreadsheet(stats):