Update opentelemetry-api/src/opentelemetry/util/_decorator.py

Co-authored-by: QuentinN42 <[email protected]>

Author: xrmx

opentelemetry-api/src/opentelemetry/util/_decorator.py

@@ -73,6 +73,14 @@ async def async_wrapper(*args: Pargs, **kwargs: Pkwargs) -> R:
 
 def _agnosticcontextmanager(
     func: "Callable[P, Iterator[R]]",
+) -> "Callable[P, _AgnosticContextManager[R]]":
+    @functools.wraps(func)
+    def helper(*args: Pargs, **kwargs: Pkwargs) -> _AgnosticContextManager[R]:
+        return _AgnosticContextManager(func, args, kwargs)
+
+    # Ingoring the type to keep the original signature of the function
+    return helper  # type: ignore[return-value]
+    func: "Callable[P, Iterator[R]]",
 ) -> "Callable[P, _AgnosticContextManager[R]]":  # type: ignore[return-value]
     @functools.wraps(func)
     def helper(*args: Pargs, **kwargs: Pkwargs) -> _AgnosticContextManager[R]: