MNT Fix incorrect source code link for wrapped objects (scikit-learn#…

…17247)
LocalSEOGuide · May 17, 2020 · 0ba19ad · 0ba19ad
1 parent 0db5f36
commit 0ba19ad
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/doc/sphinxext/github_link.py b/doc/sphinxext/github_link.py
@@ -40,12 +40,13 @@ def _linkcode_resolve(domain, info, package, url_fmt, revision):
         return
 
     class_name = info['fullname'].split('.')[0]
-    if type(class_name) != str:
-        # Python 2 only
-        class_name = class_name.encode('utf-8')
     module = __import__(info['module'], fromlist=[class_name])
     obj = attrgetter(info['fullname'])(module)
 
+    # Unwrap the object to get the correct source
+    # file in case that is wrapped by a decorator
+    obj = inspect.unwrap(obj)
+
     try:
         fn = inspect.getsourcefile(obj)
     except Exception: