From 1ac171870b89987a0a553209d5d45d3889c452ce Mon Sep 17 00:00:00 2001 From: Kush <3647166+kushsharma@users.noreply.github.com> Date: Thu, 14 Oct 2021 14:51:23 +0530 Subject: [PATCH] fix: cross entity sensor in airflow not failing if condition not met (#99) --- ext/scheduler/airflow/resources/__lib.py | 2 +- ext/scheduler/airflow2/resources/__lib.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ext/scheduler/airflow/resources/__lib.py b/ext/scheduler/airflow/resources/__lib.py index 3f1b53ce15..72692923e0 100644 --- a/ext/scheduler/airflow/resources/__lib.py +++ b/ext/scheduler/airflow/resources/__lib.py @@ -333,7 +333,7 @@ def __init__( self.window_size = window_size self._optimus_client = OptimusAPIClient(optimus_hostname) - def execute(self, context): + def poke(self, context): execution_date = context['execution_date'] execution_date_str = execution_date.strftime(self.TIMESTAMP_FORMAT) diff --git a/ext/scheduler/airflow2/resources/__lib.py b/ext/scheduler/airflow2/resources/__lib.py index d2fc4c5752..e163fe4386 100644 --- a/ext/scheduler/airflow2/resources/__lib.py +++ b/ext/scheduler/airflow2/resources/__lib.py @@ -331,7 +331,7 @@ def __init__( self.window_size = window_size self._optimus_client = OptimusAPIClient(optimus_hostname) - def execute(self, context): + def poke(self, context): execution_date = context['execution_date'] execution_date_str = execution_date.strftime(self.TIMESTAMP_FORMAT)