ci: Update release.yml (#11698)

* ci: Update release.yml Signed-off-by: oliver könig <okoenig@nvidia.com> * Update release.yml Signed-off-by: oliver könig <okoenig@nvidia.com> --------- Signed-off-by: oliver könig <okoenig@nvidia.com>
NVIDIA · Dec 21, 2024 · 17e7101 · 17e7101
1 parent 910c56a
commit 17e7101
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -20,6 +20,10 @@ on:
         description: Ref (SHA or branch name) to release
         required: true
         type: string
+      version-bump-branch:
+        description: Branch for version bump
+        required: true
+        type: string
       dry-run:
         description: Do not publish a wheel and GitHub release.
         required: true
@@ -28,7 +32,7 @@ on:
 
 jobs: 
   release:
-    uses: NVIDIA/NeMo-FW-CI-templates/.github/workflows/_release_library.yml@v0.18.1
+    uses: NVIDIA/NeMo-FW-CI-templates/.github/workflows/_release_library.yml@v0.18.2
     with:
       release-ref: ${{ inputs.release-ref }}
       image-name: nemo_container
@@ -41,6 +45,7 @@ jobs:
       container-workdir: /workspace
       library-name: Neural Modules
       dry-run: ${{ inputs.dry-run }}
+      version-bump-branch: ${{ inputs.version-bump-branch }}
     secrets:
       TWINE_USERNAME: ${{ secrets.TWINE_USERNAME }}
       TWINE_PASSWORD: ${{ secrets.TWINE_PASSWORD }}