PR 3004 Suggested Fixes As per PR 3000 updating missing #include statements (IWYU)

[MillaFleurs]

Proposed changes

Updated missing include statements. Used IWYU for linting and reviewed the output.

Checklist

Put an x in the boxes that apply.

• [ X] I have read the CONTRIBUTING document
• [X ] I have run pre-commit run --all-files to format my code / installed pre-commit prior to committing changes
• [X ] I have added tests that prove my fix is effective or that my feature works
• [X ] I have updated the necessary documentation (if needed)

[awni]

Thanks for the contribution. But as per the discussion in #3000 we don't plan to add this.