Add copy button for code in docs (pyg-team#489)

kp-forks · Jan 18, 2025 · fc7c7c3 · fc7c7c3
1 parent 1456fab
commit fc7c7c3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -20,6 +20,7 @@
     'sphinx.ext.mathjax',
     'sphinx.ext.napoleon',
     'sphinx.ext.viewcode',
+    'sphinx_copybutton',
     'pyg',
 ]
 
@@ -45,6 +46,9 @@
     'xgboost': ('https://xgboost.readthedocs.io/en/stable/', None),
 }
 
+copybutton_prompt_text = r">>> |\.\.\. "
+copybutton_prompt_is_regexp = True
+
 
 def setup(app):
     def rst_jinja_render(app, _, source):