From 51b2dcd59453cb3d9e89418424d33acbc143a427 Mon Sep 17 00:00:00 2001
From: Tim Penhey <tim@penhey.net>
Date: Thu, 18 Nov 2021 18:02:56 +1300
Subject: [PATCH] Add the __enter__ and __exit__ methods to Session.

---
 sqlalchemy-stubs/orm/session.pyi | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/sqlalchemy-stubs/orm/session.pyi b/sqlalchemy-stubs/orm/session.pyi
index af3cbed..d39b5d6 100644
--- a/sqlalchemy-stubs/orm/session.pyi
+++ b/sqlalchemy-stubs/orm/session.pyi
@@ -45,6 +45,8 @@ class Session(_SessionClassMethods):
                  binds: Optional[Any] = ..., extension: Optional[Any] = ...,
                  info: Optional[Any] = ..., query_cls: Any = ...) -> None: ...
     connection_callable: Any = ...
+    def __enter__(self): ...
+    def __exit__(self, type, value, traceback): ...
     @property
     def info(self): ...
     def begin(self, subtransactions: bool = ..., nested: bool = ...) -> SessionTransaction: ...