kubernetes executor cleanup_stuck_queued_tasks optimization

apache · Aug 15, 2024 · 0b1d4a5 · 0b1d4a5
1 parent b60cb4e
commit 0b1d4a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/airflow/providers/cncf/kubernetes/executors/kubernetes_executor.py b/airflow/providers/cncf/kubernetes/executors/kubernetes_executor.py
@@ -177,7 +177,7 @@ def _make_safe_label_value(self, input_value: str | datetime) -> str:
             return pod_generator.datetime_to_label_safe_datestring(input_value)
         return pod_generator.make_safe_label_value(input_value)
 
-    def get_pod_labels_combined_str_to_pod_map(self):
+    def get_pod_labels_combined_str_to_pod_map(self) -> dict[str, k8s.V1Pod]:
         """
         List the worker pods owned by this scheduler and create a map containing pod combined labels search str -> pod.