xauravww opened a new issue, #45940: URL: https://github.com/apache/airflow/issues/45940
### Apache Airflow version 2.10.4 ### If "Other Airflow 2 version" selected, which one? _No response_ ### What happened? OOM protection in Pinot relies on tracking resource usage of threads that execute a query. Currently, resource accounting is only supported for Single-Stage queries. This feature adds instrumentation for the Multi-Stage query engine, allowing automatic OOM protection as resource usage is accounted for in the same data structures checked by the OOM protection thread. ### What you think should happen instead? _No response_ ### How to reproduce NA ### Operating System NA ### Versions of Apache Airflow Providers _No response_ ### Deployment Official Apache Airflow Helm Chart ### Deployment details _No response_ ### Anything else? _No response_ ### Are you willing to submit PR? - [x] Yes I am willing to submit a PR! ### Code of Conduct - [x] I agree to follow this project's [Code of Conduct](https://github.com/apache/airflow/blob/main/CODE_OF_CONDUCT.md) -- 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: commits-unsubscr...@airflow.apache.org.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org