Wrap patched function

maxzheng · Sep 23, 2023 · fa1de53 · fa1de53
1 parent fdad298
commit fa1de53
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/aiohttp_requests/__init__.py b/aiohttp_requests/__init__.py
@@ -3,9 +3,10 @@
 
 # Patch ClientResponse.read to release immediately after read so we don't need to worry about that / use context manager
 _read_only = aiohttp.client_reqrep.ClientResponse.read
-async def _read_and_release(self):  # noqa
+@functools.wraps(_read_only)  # noqa
+async def _read_and_release(self, *args, **kwargs):
     try:
-        data = await _read_only(self)
+        data = await _read_only(self, *args, **kwargs)
     finally:
         self.close()