Fix watcher kube_client_request_args shadowing

apache · Jan 10, 2025 · 3177e52 · 3177e52
1 parent db739d9
commit 3177e52
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/providers/src/airflow/providers/cncf/kubernetes/executors/kubernetes_executor_utils.py b/providers/src/airflow/providers/cncf/kubernetes/executors/kubernetes_executor_utils.py
@@ -147,10 +147,12 @@ def _run(
         # For info about k8s timeout settings see
         # https://github.com/kubernetes-client/python/blob/v29.0.0/examples/watch/timeout-settings.md
         # and https://github.com/kubernetes-client/python/blob/v29.0.0/kubernetes/client/api_client.py#L336-L339
-        client_timeout = 30
-        server_conn_timeout = 3600
-        kwargs["_request_timeout"] = client_timeout
-        kwargs["timeout_seconds"] = server_conn_timeout
+        if "_request_timeout" not in kwargs:
+            client_timeout = 30
+            kwargs["_request_timeout"] = client_timeout
+        if "timeout_seconds" not in kwargs:
+            server_conn_timeout = 3600
+            kwargs["timeout_seconds"] = server_conn_timeout
 
         logical_date_key = get_logical_date_key()
         for event in self._pod_events(kube_client=kube_client, query_kwargs=kwargs):