[BUG] Fix redirecting traceback from stderr to a file

[ikondov]

Closes #562

Summary

Major changes:

• fix 1: Return the file handles from context manager
• fix 2: Pass the stderr file handle as argument so that the traceback function is called with it

Todos

Checklist

• Google format doc strings added. Check with ruff.
• Type annotations included. Check with mypy.
• Tests added for new features/fixes.
• If applicable, new classes/functions/modules have duecredit @due.dcite decorators to reference relevant papers by DOI (example)

Tip: Install pre-commit hooks to auto-check types and linting before every commit:

pip install -U pre-commit
pre-commit install