This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new b2216e6114f Reflect new base_container_status_polling_interval in the
TaskDecoratorCollection type (#47833)
b2216e6114f is described below
commit b2216e6114f13b0f5eae45904d0b636e75a8438e
Author: Balthazar Rouberol <[email protected]>
AuthorDate: Thu Mar 20 19:51:34 2025 +0100
Reflect new base_container_status_polling_interval in the
TaskDecoratorCollection type (#47833)
This is a follow up to https://github.com/apache/airflow/pull/47774,
addind the type of the new `base_container_status_polling_interval` argument
to the `TaskDecoratorCollection` type, to update the type autocompletion
when using the `@kubernetes` decorator.
Signed-off-by: Balthazar Rouberol <[email protected]>
---
airflow/decorators/__init__.pyi | 3 +++
1 file changed, 3 insertions(+)
diff --git a/airflow/decorators/__init__.pyi b/airflow/decorators/__init__.pyi
index 34f110983cb..ea12e95b68c 100644
--- a/airflow/decorators/__init__.pyi
+++ b/airflow/decorators/__init__.pyi
@@ -569,6 +569,7 @@ class TaskDecoratorCollection:
configmaps: list[str] | None = None,
skip_on_exit_code: int | Container[int] | None = None,
base_container_name: str | None = None,
+ base_container_status_polling_interval: float = ...,
deferrable: bool = ...,
poll_interval: float = ...,
log_pod_spec_on_failure: bool = ...,
@@ -669,6 +670,8 @@ class TaskDecoratorCollection:
will appear as part of this task's logs if get_logs is True.
Defaults to None. If None,
will consult the class variable BASE_CONTAINER_NAME (which
defaults to "base") for the base
container name to use.
+ :param base_container_status_polling_interval: Polling period in
seconds to check for the pod base
+ container status.
:param deferrable: Run operator in the deferrable mode.
:param poll_interval: Polling period in seconds to check for the
status. Used only in deferrable mode.
:param log_pod_spec_on_failure: Log the pod's specification if a
failure occurs
