ci: Update pyproject.toml with dependabot (#171)

Signed-off-by: oliver könig <okoenig@nvidia.com>

Author: ko3n1g

.github/workflows/_update_dependencies.yml

@@ -33,7 +33,7 @@ jobs:
     runs-on: ubuntu-latest
     outputs:
       source-branch: ${{ steps.merge-branch.outputs.source-branch }}
-      bump-branch: bump-ci-container-${{ steps.ref.outputs.date }}
+      bump-branch: bump-ci-container-${{ steps.ref.outputs.date }}-${{ steps.merge-branch.outputs.source-branch }}
       date: ${{ steps.ref.outputs.date }}
     steps:
       - uses: actions/checkout@v2
@@ -71,14 +71,14 @@ jobs:
           git push -u origin $SOURCE_BRANCH
 
   mcore:
-    uses: NVIDIA-NeMo/FW-CI-templates/.github/workflows/_bump_yamlfile.yml@v0.36.1
+    uses: NVIDIA-NeMo/FW-CI-templates/.github/workflows/_bump_tomlfile.yml@v0.40.0
     needs: [update-branch]
     name: Bump Megatron-LM
     with:
       source-repository: NVIDIA/Megatron-LM
       source-ref: ${{ inputs.mcore-target-branch }}
-      yaml-path: '."vcs-dependencies"."megatron-lm".ref'
-      file: docker/manifest.json
+      toml-path: tool.uv.sources.megatron-core.rev
+      file: pyproject.toml
       base-branch: ${{ needs.update-branch.outputs.source-branch }}
       pr-branch: ${{ needs.update-branch.outputs.bump-branch }}
       pr-title: "chore(🤖): Bump dependencies (${{ needs.update-branch.outputs.date }})"
@@ -89,14 +89,14 @@ jobs:
       GPG_PASSPHRASE: ${{ secrets.SSH_PWD }}
 
   nemo:
-    uses: NVIDIA-NeMo/FW-CI-templates/.github/workflows/_bump_yamlfile.yml@v0.36.1
+    uses: NVIDIA-NeMo/FW-CI-templates/.github/workflows/_bump_tomlfile.yml@v0.40.0
     needs: [update-branch, mcore]
     name: Bump NeMo
     with:
       source-repository: NVIDIA/NeMo
       source-ref: ${{ inputs.nemo-target-branch }}
-      yaml-path: '."vcs-dependencies"."nemo".ref'
-      file: docker/manifest.json
+      toml-path: tool.uv.sources.nemo-toolkit.rev
+      file: pyproject.toml
       base-branch: ${{ needs.update-branch.outputs.source-branch }}
       pr-branch: ${{ needs.update-branch.outputs.bump-branch }}
       pr-title: "chore(🤖): Bump dependencies (${{ needs.update-branch.outputs.date }})"
@@ -109,7 +109,7 @@ jobs:
   update-lockfile:
     environment: nemo-ci
     runs-on: linux-amd64-cpu16
-    needs: [nemo]
+    needs: [nemo, update-branch]
     steps:
       - name: Install Azure CLI
         run: curl -sL https://aka.ms/InstallAzureCLIDeb | sudo bash
@@ -139,7 +139,7 @@ jobs:
       - name: Upload lock file
         uses: actions/upload-artifact@v4
         with:
-          name: lock-file
+          name: lock-file-${{ needs.update-branch.outputs.bump-branch }}
           path: uv.lock
 
   create-pr:
@@ -167,7 +167,7 @@ jobs:
       - name: Download lock file
         uses: actions/download-artifact@v4
         with:
-          name: lock-file
+          name: lock-file-${{ needs.update-branch.outputs.bump-branch }}
 
       - name: Commit changes
         run: |

docker/Dockerfile.ci

@@ -32,8 +32,6 @@ ENV UV_LINK_MODE=copy
 # Install all
 WORKDIR /workspace
 ARG INFERENCE_FRAMEWORK
-ARG NEMO_REF
-ARG MCORE_REF
 RUN --mount=type=bind,source=pyproject.toml,target=/workspace/pyproject.toml \
     --mount=type=bind,source=uv.lock,target=uv.lock \
     --mount=type=bind,source=nemo_export/__init__.py,target=/workspace/nemo_export/__init__.py \