DOC: Fixes line numbers on classes decorated with set_module (#28629)

aureliobarbosa · aureliobarbosa · commit 6ed0ece90bb8 · 2025-05-14T11:54:05.000-03:00
diff --git a/doc/source/conf.py b/doc/source/conf.py
@@ -539,14 +539,14 @@ def linkcode_resolve(domain, info):
     fn = None
     lineno = None
 
-    # Make a poor effort at linking C extension types
-    if isinstance(obj, type) and obj.__module__ == 'numpy':
-        fn = _get_c_source_file(obj)
+    if isinstance(obj, type):
+        # Make a poor effort at linking C extension types
+        if obj.__module__ == 'numpy':
+            fn = _get_c_source_file(obj)
 
-    # This can be removed when removing the decorator set_module. Fix issue #28629
-    if hasattr(obj, '_module_file'):
-        fn = obj._module_file
-        fn = relpath(fn, start=dirname(numpy.__file__))
+        # This can be removed when removing the decorator set_module. Fix issue #28629
+        if hasattr(obj, '_module_source'):
+            obj.__module__, obj._module_source = obj._module_source, obj.__module__
 
     if fn is None:
         try:
@@ -573,6 +573,9 @@ def linkcode_resolve(domain, info):
     else:
         linespec = ""
 
+    if isinstance(obj, type) and hasattr(obj, '_module_source'):
+        obj.__module__, obj._module_source = obj._module_source, obj.__module__
+
     if 'dev' in numpy.__version__:
         return f"https://github.com/numpy/numpy/blob/main/numpy/{fn}{linespec}"
     else:
diff --git a/numpy/_utils/__init__.py b/numpy/_utils/__init__.py
@@ -9,7 +9,6 @@
 """
 
 import functools
-import sys
 import warnings
 from ._convertions import asunicode, asbytes
 
@@ -29,8 +28,8 @@ def decorator(func):
         if module is not None:
             if isinstance(func, type):
                 try:
-                    func._module_file = sys.modules.get(func.__module__).__file__
-                except (AttributeError, KeyError):
+                    func._module_source = func.__module__
+                except (AttributeError):
                     pass
 
             func.__module__ = module
