Additional PR feedback

Author: strawgate

key-value/key-value-aio/src/key_value/aio/stores/base.py

@@ -474,9 +474,12 @@ async def __aexit__(
     ) -> None:
         # Close the exit stack, which handles all cleanup
         if self._exit_stack_entered:
-            await self._exit_stack.__aexit__(exc_type, exc_value, traceback)
+            result = await self._exit_stack.__aexit__(exc_type, exc_value, traceback)
             self._exit_stack_entered = False
 
+            return result
+        return None
+
     async def close(self) -> None:
         # Close the exit stack if it has been entered
         if self._exit_stack_entered:

key-value/key-value-sync/src/key_value/sync/code_gen/stores/base.py

@@ -444,9 +444,12 @@ def __enter__(self) -> Self:
     def __exit__(self, exc_type: type[BaseException] | None, exc_value: BaseException | None, traceback: TracebackType | None) -> None:
         # Close the exit stack, which handles all cleanup
         if self._exit_stack_entered:
-            self._exit_stack.__exit__(exc_type, exc_value, traceback)
+            result = self._exit_stack.__exit__(exc_type, exc_value, traceback)
             self._exit_stack_entered = False
 
+            return result
+        return None
+
     def close(self) -> None:
         # Close the exit stack if it has been entered
         if self._exit_stack_entered: