Ensure shortened numpy and galois references

mhostetter · Nov 20, 2023 · 298c1d6 · 298c1d6
1 parent 48fca42
commit 298c1d6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/docs/conf.py b/docs/conf.py
@@ -13,6 +13,7 @@
 
 import inspect
 import os
+import re
 import sys
 
 sys.path.insert(0, os.path.abspath("."))
@@ -440,7 +441,8 @@ def modify_type_hints(signature):
     See https://github.com/jbms/sphinx-immaterial/issues/161
     """
     if signature:
-        signature = signature.replace("np", "~numpy")
+        signature = re.sub(r"(?<!~)np\.", "~numpy.", signature)
+        signature = re.sub(r"(?<!~)galois\.", "~galois.", signature)
     return signature