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)