updates pipeline_timeout CloudDataFusionStartPipelineOperator (#18773)

apache · Nov 26, 2021 · a192cec · a192cec
1 parent 20dc5b9
commit a192cec
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/airflow/providers/google/cloud/operators/datafusion.py b/airflow/providers/google/cloud/operators/datafusion.py
@@ -802,7 +802,7 @@ def __init__(
  runtime_args: Optional[Dict[str, Any]] = None,
  success_states: Optional[List[str]] = None,
  namespace: str = "default",
- pipeline_timeout: int = 10 * 60,
+ pipeline_timeout: int = 5 * 60,
  project_id: Optional[str] = None,
  api_version: str = "v1beta1",
  gcp_conn_id: str = "google_cloud_default",
@@ -823,13 +823,12 @@ def __init__(
  self.delegate_to = delegate_to
  self.impersonation_chain = impersonation_chain
  self.asynchronous = asynchronous
+ self.pipeline_timeout = pipeline_timeout
 
  if success_states:
  self.success_states = success_states
- self.pipeline_timeout = pipeline_timeout
  else:
  self.success_states = SUCCESS_STATES + [PipelineStates.RUNNING]
- self.pipeline_timeout = 5 * 60
 
  def execute(self, context: dict) -> str:
  hook = DataFusionHook(