fix: allow wrapped function

TexteaInc · Jul 28, 2024 · 1432113 · 1432113
1 parent 3eba23c
commit 1432113
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/backend/funix/util/module.py b/backend/funix/util/module.py
@@ -104,7 +104,11 @@ def handle_module(
         is_cls = isclass(module_member)
         if is_func or is_cls:
             if getsourcefile_safe(module_member) != module.__file__:
-                continue
+                if hasattr(module_member, "__wrapped__"):
+                    if getsourcefile_safe(module_member.__wrapped__) != module.__file__:
+                        continue
+                else:
+                    continue
             in_funix = (
                 decorator.object_is_handled(app, id(module_member))
                 or id(module_member) in hint.custom_cls_ids