Cleanup CommandCursor init

mongodb · Jun 14, 2024 · 3d95075 · 3d95075
1 parent c27a36a
commit 3d95075
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/pymongo/asynchronous/command_cursor.py b/pymongo/asynchronous/command_cursor.py
@@ -81,8 +81,8 @@ def __init__(
         self._explicit_session = explicit_session
         self._killed = self._id == 0
         self._comment = comment
-        if _IS_SYNC and self._killed:
-            self._end_session(True)  # type: ignore[unused-coroutine]
+        if self._killed:
+            self._end_session()
 
         if "ns" in cursor_info:  # noqa: SIM401
             self._ns = cursor_info["ns"]
@@ -234,9 +234,9 @@ async def _die_lock(self) -> None:
             self._session = None
         self._sock_mgr = None
 
-    async def _end_session(self, synchronous: bool) -> None:
+    def _end_session(self) -> None:
         if self._session and not self._explicit_session:
-            await self._session._end_session(lock=synchronous)
+            self._session._end_implicit_session()
             self._session = None
 
     async def close(self) -> None:

diff --git a/pymongo/synchronous/command_cursor.py b/pymongo/synchronous/command_cursor.py
@@ -81,8 +81,8 @@ def __init__(
         self._explicit_session = explicit_session
         self._killed = self._id == 0
         self._comment = comment
-        if _IS_SYNC and self._killed:
-            self._end_session(True)  # type: ignore[unused-coroutine]
+        if self._killed:
+            self._end_session()
 
         if "ns" in cursor_info:  # noqa: SIM401
             self._ns = cursor_info["ns"]
@@ -234,9 +234,9 @@ def _die_lock(self) -> None:
             self._session = None
         self._sock_mgr = None
 
-    def _end_session(self, synchronous: bool) -> None:
+    def _end_session(self) -> None:
         if self._session and not self._explicit_session:
-            self._session._end_session(lock=synchronous)
+            self._session._end_implicit_session()
             self._session = None
 
     def close(self) -> None: