ContextManager - missing 3 required positional arguments: 'exc_type', 'exc_val', and 'exc_tb'

[Tusenka]

Given:

def get_session():
    class _SessionContext:
        def __init__(self):
            print("init")

        def __enter__(self):
            return "enter"

        def __exit__(self, exc_type, exc_val, exc_tb):
            print("exit")
    return _SessionContext()

@codon.jit(pyvars=['get_session'])
def main():
    with get_session() as session:
        print(session)

main()

Expected:
Success execution

Actual:
Error is missing 3 required positional arguments: 'exc_type', 'exc_val', and 'exc_tb'