Skip to content

Detect Spark driver completion by container state when tracking via k8s API#68048

Open
karenbraganz wants to merge 12 commits into
apache:mainfrom
karenbraganz:spark-container-status
Open

Detect Spark driver completion by container state when tracking via k8s API#68048
karenbraganz wants to merge 12 commits into
apache:mainfrom
karenbraganz:spark-container-status

Conversation

@karenbraganz

@karenbraganz karenbraganz commented Jun 5, 2026

Copy link
Copy Markdown
Collaborator

This PR tracks Spark job completion by container state instead of pod phase when track_driver_via_k8s_api=True. Sometimes the pod continues to run even after the driver container completes due to other sidecar containers. This PR makes driver completion detection more accurate by examining the container itself.

@karenbraganz

karenbraganz commented Jun 5, 2026

Copy link
Copy Markdown
Collaborator Author

I still need to test this out and write unit tests.

@karenbraganz
karenbraganz requested a review from amoghrajesh June 9, 2026 16:39
@karenbraganz

Copy link
Copy Markdown
Collaborator Author

This has passed all unit tests as well as a manual test that I ran.

Comment thread providers/apache/spark/src/airflow/providers/apache/spark/hooks/spark_submit.py Outdated
@karenbraganz
karenbraganz requested a review from uranusjr July 21, 2026 13:48
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

For spark operator with track_driver_via_k8s_api, detect driver completion by container status rather than pod phase

2 participants