From 33738166663de1151fbd50ee0fafadcd17fe48ee Mon Sep 17 00:00:00 2001
From: Alvaro Bartolome <36760800+alvarobartt@users.noreply.github.com>
Date: Fri, 11 Oct 2024 10:36:03 +0200
Subject: [PATCH] Fix `docs/scripts/auto-generate-examples.py` (#110)

Hot-fix `docs/scripts/auto-generate-examples.py` script as the
`GITHUB_BRANCH` environment variable is empty, so rollback to use `main`
by default which is not ideal either since the PR examples won't have
working images, but lets tackle this on another PR later on.
---
 docs/scripts/auto-generate-examples.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/docs/scripts/auto-generate-examples.py b/docs/scripts/auto-generate-examples.py
index aba7e63b..4f87a105 100644
--- a/docs/scripts/auto-generate-examples.py
+++ b/docs/scripts/auto-generate-examples.py
@@ -1,8 +1,6 @@
 import os
 import re
 
-GITHUB_BRANCH = os.getenv("GITHUB_HEAD_REF", "main")
-
 
 def process_readme_files():
     print("Processing README.md files from examples/gke and examples/cloud-run...")
@@ -37,21 +35,21 @@ def process_file(root, file, dir):
     # Replace image and link paths
     content = re.sub(
         r"\(\./(imgs|assets)/([^)]*\.png)\)",
-        rf"(https://raw.githubusercontent.com/huggingface/Google-Cloud-Containers/{GITHUB_BRANCH}/"
+        r"(https://raw.githubusercontent.com/huggingface/Google-Cloud-Containers/main/"
         + root
         + r"/\1/\2)",
         content,
     )
     content = re.sub(
         r"\(\.\./([^)]+)\)",
-        rf"(https://github.com/huggingface/Google-Cloud-Containers/tree/{GITHUB_BRANCH}/examples/"
+        r"(https://github.com/huggingface/Google-Cloud-Containers/tree/main/examples/"
         + dir
         + r"/\1)",
         content,
     )
     content = re.sub(
         r"\(\.\/([^)]+)\)",
-        rf"(https://github.com/huggingface/Google-Cloud-Containers/tree/{GITHUB_BRANCH}/"
+        r"(https://github.com/huggingface/Google-Cloud-Containers/tree/main/"
         + root
         + r"/\1)",
         content,