Use READTHEDOCS_VERSION_NAME

dmlc · Jun 4, 2024 · 462cb2c · 462cb2c
1 parent e6fd3f1
commit 462cb2c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/doc/conf.py b/doc/conf.py
@@ -61,9 +61,14 @@ def run_doxygen():
 
 def build_jvm_docs():
     """Build docs for the JVM packages"""
-    git_branch = os.getenv("READTHEDOCS_GIT_IDENTIFIER", default=None)
+    git_branch = os.getenv("READTHEDOCS_VERSION_NAME", default=None)
     print(f"git_branch = {git_branch}")
 
+    if git_branch == "latest":
+        git_branch = "master"
+    elif git_branch == "stable":
+        git_branch = f"release_{version}"
+
     def try_fetch_jvm_doc(branch):
         """
         Attempt to fetch JVM docs for a given branch.