diff --git a/ddtrace/contrib/asyncio/patch.py b/ddtrace/contrib/asyncio/patch.py index a1bad51671ce..257576709b1f 100644 --- a/ddtrace/contrib/asyncio/patch.py +++ b/ddtrace/contrib/asyncio/patch.py @@ -31,10 +31,10 @@ def unpatch(): if getattr(asyncio, "_datadog_patch", False): setattr(asyncio, "_datadog_patch", False) - if sys.version_info < (3, 7, 0): - _u(asyncio.BaseEventLoop, "create_task") + if sys.version_info < (3, 7, 0): + _u(asyncio.BaseEventLoop, "create_task") - # also unpatch event loop if not inheriting the already unwrapped create_task from BaseEventLoop - loop = asyncio.get_event_loop() - if isinstance(loop.create_task, ObjectProxy): - _u(loop, "create_task") + # also unpatch event loop if not inheriting the already unwrapped create_task from BaseEventLoop + loop = asyncio.get_event_loop() + if isinstance(loop.create_task, ObjectProxy): + _u(loop, "create_task") diff --git a/ddtrace/utils/wrappers.py b/ddtrace/utils/wrappers.py index 68cd812e7b64..c4dbb66000a8 100644 --- a/ddtrace/utils/wrappers.py +++ b/ddtrace/utils/wrappers.py @@ -27,9 +27,8 @@ def iswrapped(obj, attr=None): def unwrap(obj, attr): # type: (Any, str) -> None - f = getattr(obj, attr, None) - if f and isinstance(f, wrapt.ObjectProxy) and hasattr(f, "__wrapped__"): - setattr(obj, attr, f.__wrapped__) + f = getattr(obj, attr) + setattr(obj, attr, f.__wrapped__) @deprecated("`wrapt` library is used instead", version="1.0.0")