From a7d009f80142386dc39dd23ac6d7076f0473bd0b Mon Sep 17 00:00:00 2001
From: isra17 <isra017@gmail.com>
Date: Wed, 18 Sep 2019 13:27:09 -0400
Subject: [PATCH] Add type to Session begin and begin_nested

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

diff --git a/sqlalchemy-stubs/orm/session.pyi b/sqlalchemy-stubs/orm/session.pyi
index 0828ea3..d5f3f47 100644
--- a/sqlalchemy-stubs/orm/session.pyi
+++ b/sqlalchemy-stubs/orm/session.pyi
@@ -42,8 +42,8 @@ class Session(_SessionClassMethods):
     connection_callable: Any = ...
     @property
     def info(self): ...
-    def begin(self, subtransactions: bool = ..., nested: bool = ...): ...
-    def begin_nested(self): ...
+    def begin(self, subtransactions: bool = ..., nested: bool = ...) -> SessionTransaction: ...
+    def begin_nested(self) -> SessionTransaction: ...
     def rollback(self): ...
     def commit(self): ...
     def prepare(self): ...