Fix return type signature for context managers

[lpsinger]

Set the return type to Self so that the type annotations are correct for subclasses.

What

Checklist

• Contains customer facing changes? Including API/behavior changes
• Did you add sufficient unit test and/or integration test coverage for this PR?
  • If not, please explain why it is not required

References

JIRA:

Test & Review

Open questions / Follow-ups

[confluent-cla-assistant]

🎉 All Contributor License Agreements have been signed. Ready to merge.
✅ lpsinger
_{Please push an empty commit if you would like to re-run the checks to verify CLA status for all contributors.}