Preserve decorated function's return type hint

[collindutter]

• [x] I have read and agree to the contributing guidelines for submitting new pull requests.

Describe your changes

Given:

Before: After:

Issue ticket number and link

NA

[codecov[bot]]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

:loudspeaker: Thoughts on this report? Let us know!