Add back reftarget change for cudf.Index

vyasr · Jan 30, 2024 · 71d87d5 · 71d87d5
1 parent 6d07cc2
commit 71d87d5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/docs/cudf/source/conf.py b/docs/cudf/source/conf.py
@@ -454,6 +454,12 @@ def on_missing_reference(app, env, node, contnode):
                 _prefixed_domain_objects[f"{prefix}{name}"] = name
 
     reftarget = node.get("reftarget")
+    if reftarget == "cudf.core.index.Index":
+        # We don't exposed docs for `cudf.core.index.Index`
+        # hence we would want the docstring & mypy references to
+        # use `cudf.Index`
+        node["reftarget"] = "cudf.Index"
+        return contnode
     if "namespacecudf" in reftarget:
         node["reftarget"] = "cudf"
         return contnode