From feddedf97d54503635774ae84b722160aa6612a8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ole=20Christian=20Langfj=C3=A6ran?= Date: Fri, 15 Mar 2024 10:15:25 +0100 Subject: [PATCH] Add use of timeout --- .../providers/google/gcs_trigger_dag_configuration.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/unaflow/dags/trigger_dag/providers/google/gcs_trigger_dag_configuration.py b/unaflow/dags/trigger_dag/providers/google/gcs_trigger_dag_configuration.py index ff2c19b..19a7fcb 100644 --- a/unaflow/dags/trigger_dag/providers/google/gcs_trigger_dag_configuration.py +++ b/unaflow/dags/trigger_dag/providers/google/gcs_trigger_dag_configuration.py @@ -66,7 +66,8 @@ def create_sensor(self) -> BaseSensorOperator: prefix=self.prefix, google_cloud_conn_id=self.gcp_conn_id, poke_interval=self.poke_interval, - mode=self.mode + mode=self.mode, + timeout=self.timeout, ) def create_downstream_sensor(self) -> Union[TaskMixin, Sequence[TaskMixin]]: