-
Notifications
You must be signed in to change notification settings - Fork 251
Open
Labels
customer requestAn Astronomer customer made requested thisAn Astronomer customer made requested thisexecution:watcherRelated to Watcher execution modeRelated to Watcher execution mode
Description
Summary
In watcher mode, we currently rely on ExecutionConfig.setup_operator_args (PR #2133) to send the producer task to a high-resource queue while sensors stay on a lightweight queue. When sensors retry (and run _fallback_to_local_run), the retry still executes on the sensor queue. We aim to validate and document a supported approach for routing sensor retries to the producer queue using an Airflow cluster policy.
What should happen
- A concrete recipe lives in docs (and potentially tests) showing how to use task_instance_mutation_hook to flip
ti.queueto the producer queue on retries (e.g., when watcher sensors run_fallback_to_local_run). - We confirm Cosmos works with this configuration (no regressions from the hook) and that retries succeed when the producer queue is used.
- Optional: consider adding a helper in Cosmos (or a sample snippet) so users don’t have to hand-roll the hook.
Metadata
Metadata
Assignees
Labels
customer requestAn Astronomer customer made requested thisAn Astronomer customer made requested thisexecution:watcherRelated to Watcher execution modeRelated to Watcher execution mode