🎨 Clarify types / args

Signed-off-by: ff137 <[email protected]>
didx-xyz · Feb 6, 2025 · e3d639c · e3d639c
1 parent 39caf89
commit e3d639c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/acapy_agent/askar/profile.py b/acapy_agent/askar/profile.py
@@ -201,13 +201,15 @@ def session(
         else:
             return AskarProfileSession(self, is_txn=False, context=context)
 
-    def transaction(self, context: Optional[InjectionContext] = None) -> ProfileSession:
+    def transaction(
+        self, context: Optional[InjectionContext] = None
+    ) -> "AskarProfileSession":
         """Start a new interactive session with commit and rollback support.
 
         If the current backend does not support transactions, then commit
         and rollback operations of the session will not have any effect.
         """
-        return AskarProfileSession(self, True, context=context)
+        return AskarProfileSession(self, is_txn=True, context=context)
 
     async def close(self):
         """Close the profile instance."""

diff --git a/acapy_agent/askar/profile_anon.py b/acapy_agent/askar/profile_anon.py
@@ -190,7 +190,7 @@ def transaction(
         If the current backend does not support transactions, then commit
         and rollback operations of the session will not have any effect.
         """
-        return AskarAnoncredsProfileSession(self, True, context=context)
+        return AskarAnoncredsProfileSession(self, is_txn=True, context=context)
 
     async def close(self):
         """Close the profile instance."""