docs: Bump HPU ref 1.6.0 (#20026)

--------- Co-authored-by: jerome-habana <[email protected]> Co-authored-by: Jirka Borovec <[email protected]> Co-authored-by: awaelchli <[email protected]>
Lightning-AI · Jun 28, 2024 · fa5af16 · fa5af16
1 parent aa2da72
commit fa5af16
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docs/source-pytorch/conf.py b/docs/source-pytorch/conf.py
@@ -92,14 +92,14 @@ def _load_py_module(name: str, location: str) -> ModuleType:
 assist_local.AssistantCLI.pull_docs_files(
     gh_user_repo="Lightning-AI/lightning-Habana",
     target_dir="docs/source-pytorch/integrations/hpu",
-    checkout="refs/tags/1.5.0",
+    checkout="refs/tags/1.6.0",
 )
 # the HPU also need some images
-URL_RAW_DOCS_GRAPHCORE = "https://raw.githubusercontent.com/Lightning-AI/lightning-Habana/1.5.0/docs/source"
+URL_RAW_DOCS_HABANA = "https://raw.githubusercontent.com/Lightning-AI/lightning-Habana/1.5.0/docs/source"
 for img in ["_images/HPUProfiler.png", "_images/IGP.png"]:
     img_ = os.path.join(_PATH_HERE, "integrations", "hpu", img)
     os.makedirs(os.path.dirname(img_), exist_ok=True)
-    urllib.request.urlretrieve(f"{URL_RAW_DOCS_GRAPHCORE}/{img}", img_)
+    urllib.request.urlretrieve(f"{URL_RAW_DOCS_HABANA}/{img}", img_)
 
 # Copy strategies docs as single pages
 assist_local.AssistantCLI.pull_docs_files(