Fix docstrings

gerlero · Sep 23, 2024 · 065ed61 · 065ed61
1 parent cbdf671
commit 065ed61
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/foamlib/_cases/_util.py b/foamlib/_cases/_util.py
@@ -1,3 +1,4 @@
+import functools
 from types import TracebackType
 from typing import Any, AsyncContextManager, Callable, Optional, Type
 
@@ -24,4 +25,8 @@ async def __aexit__(
 def awaitableasynccontextmanager(
     cm: Callable[..., "AsyncContextManager[Any]"],
 ) -> Callable[..., _AwaitableAsyncContextManager]:
-    return lambda *args, **kwargs: _AwaitableAsyncContextManager(cm(*args, **kwargs))
+    @functools.wraps(cm)
+    def f(*args: Any, **kwargs:Any) -> _AwaitableAsyncContextManager:
+        return _AwaitableAsyncContextManager(cm(*args, **kwargs))
+
+    return f