moomindani opened a new pull request, #68924: URL: https://github.com/apache/airflow/pull/68924
Fix `DatabricksWorkflowTaskGroup` so that upstream/downstream dependencies set via `>>` *before* the `with` block are correctly transferred to the internal `launch` task and leaf tasks. When `>>` was called before the `with task_group:` block, the dependency was recorded on the task group but never transferred to the `launch` task (upstream) or the leaf tasks (downstream), because those tasks did not exist yet. Transfer `self.upstream_task_ids` and `self.downstream_task_ids` in `__exit__()` after creating the internal tasks. Note: PR #67369 by @shubhgurav0590 identified the correct approach in [this comment](https://github.com/apache/airflow/pull/67369#issuecomment-2905764696) but the fix was never committed before the PR went stale. This PR implements that approach and adds downstream coverage. The commit includes Co-authored-by to preserve credit. closes: #51598 --- ##### Was generative AI tooling used to co-author this PR? - [X] Yes — Claude Code (Opus 4.8) Generated-by: Claude Code (Opus 4.8) following [the guidelines](https://github.com/apache/airflow/blob/main/contributing-docs/05_pull_requests.rst#gen-ai-assisted-contributions) -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
