From 4030e00fd39149dfedbd830da4d0b38a13dd0b57 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?oliver=20k=C3=B6nig?= <okoenig@nvidia.com>
Date: Sat, 25 Oct 2025 14:26:43 +0000
Subject: [PATCH] ci: Fix branch of approval bot
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: oliver könig <okoenig@nvidia.com>
---
 .gitlab/stages/05.publish.yml | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/.gitlab/stages/05.publish.yml b/.gitlab/stages/05.publish.yml
index c925a92d1c3..b566290e4af 100644
--- a/.gitlab/stages/05.publish.yml
+++ b/.gitlab/stages/05.publish.yml
@@ -776,7 +776,11 @@ publish:approve_merge_gate:
 
       export GH_TOKEN=$GH_TOKEN
       export REPO=NVIDIA/Megatron-LM
-      export TARGET_BRANCH="$CI_COMMIT_BRANCH"
+      if [[ "$CI_COMMIT_BRANCH" == *main* ]]; then
+        export TARGET_BRANCH="main"
+      elif [[ "$CI_COMMIT_BRANCH" == *dev* ]]; then
+        export TARGET_BRANCH="dev"
+      fi
 
       if [[ $EXIT_CODE -eq 0 ]]; then
         export STATUS="approved"