docs: fix linkcode for identify methods patched in metaclass

kmnhan · Dec 17, 2024 · 40f2fbb · 40f2fbb
1 parent da5244a
commit 40f2fbb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -94,6 +94,10 @@ def linkcode_resolve(domain, info) -> str | None:
     modname = info["module"]
     fullname = info["fullname"]
 
+    # Handle dataloader metaclass substitution
+    if fullname.endswith(".identify") and not fullname.endswith(".LoaderBase.identify"):
+        fullname = fullname.replace(".identify", "._original_identify")
+
     submod = sys.modules.get(modname)
     if submod is None:
         return None