Update dags_airflow_fede_k8sexecutor.py

interTwin-eu · Jan 29, 2024 · 4010444 · 4010444
1 parent b2891d9
commit 4010444
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/dags_airflow_fede_k8sexecutor.py b/dags_airflow_fede_k8sexecutor.py
@@ -33,6 +33,8 @@
     )
     k8s = None
 
+default_queue = "kubernetes"
+
 if k8s:
     with DAG(
         dag_id="kubernetes_executor",
@@ -56,7 +58,7 @@
         #############################################################
         @task(
             executor_config=executor_config_annotation,
-            queue="kubernetes",
+            queue=default_queue,
             task_id="task_k8s_annotation",
         )
         def test_annotation():
@@ -95,7 +97,7 @@ def test_annotation():
         ###########################################################
         @task(
             executor_config=executor_config_volume_mount,
-            queue="kubernetes",
+            queue=default_queue,
             task_id="task_k8s_volume_mount",
         )
         def test_volume_mount():
@@ -165,7 +167,7 @@ def test_volume_mount():
         ###########################################################
         @task(
             executor_config=executor_config_sidecar,
-            queue="kubernetes",
+            queue=default_queue,
             task_id="task_k8s_sidecar",
         )
         def test_sharedvolume_mount():
@@ -199,7 +201,7 @@ def test_sharedvolume_mount():
         #############################################################
         @task(
             executor_config=executor_config_label,
-            queue="kubernetes",
+            queue=default_queue,
             task_id="task_k8s_label",
         )
         def test_label():
@@ -223,7 +225,7 @@ def test_label():
         #############################################################
         @task(
             executor_config=executor_config_other_ns,
-            queue="kubernetes",
+            queue=default_queue,
             task_id="task_k8s_other_ns",
         )
         def other_namespace_task():
@@ -257,7 +259,7 @@ def other_namespace_task():
         #############################################################
         @task(
             executor_config=kube_exec_config_image,
-            queue="kubernetes",
+            queue=default_queue,
             task_id="task_k8s_image",
         )
         def image_override_task():
@@ -317,7 +319,7 @@ def image_override_task():
         #############################################################
         @task(
             executor_config=kube_exec_config_resource_limits,
-            queue="kubernetes",
+            queue=default_queue,
             task_id="task_k8s_resource_limits",
         )
         def task_with_resource_limits():