Add use of timeout

unacast · Mar 15, 2024 · feddedf · feddedf
1 parent 1b6d6d2
commit feddedf
Showing 1 changed file with 2 additions and 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
@@ -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]]: