Automatically add review label

.github/workflows/auto-assign-review-labels.yml

@@ -0,0 +1,33 @@
+# Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+
+name: Auto Assign Review Labels
+on:
+  pull_request:
+    types: [opened, ready_for_review]
+
+permissions:
+  pull-requests: write
+  contents: read
+
+jobs:
+  assign-label:
+    runs-on: ubuntu-latest
+    if: >-
+      !github.event.pull_request.draft
+      && github.repository == 'NVIDIA/Megatron-LM'
+    steps:
+      - name: Check out repository code
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.10"
+
+      - name: Install dependencies
+        run: pip install --no-cache-dir PyGithub
+
+      - name: Assign initial review label
+        env:
+          GH_TOKEN: ${{ github.token }}
+        run: python tests/test_utils/python_scripts/update_pr_labels.py --pr ${{ github.event.pull_request.number }} assign

.github/workflows/auto-swap-labels.yml

@@ -23,11 +23,9 @@ jobs:
           python-version: "3.10"
 
       - name: Install dependencies
-        run: |
-          pip install --no-cache-dir PyGithub slack-sdk
+        run: pip install --no-cache-dir PyGithub
 
-      - name: Run Auto Reminder Bot
-        run: |
-          export GH_TOKEN=${{ github.token }}
-          export PR_NUMBER=${{ github.event.pull_request.number }}
-          python tests/test_utils/python_scripts/swap_pr_labels.py
+      - name: Swap labels after approval
+        env:
+          GH_TOKEN: ${{ github.token }}
+        run: python tests/test_utils/python_scripts/update_pr_labels.py --pr ${{ github.event.pull_request.number }} swap