cmd

Lightning-AI · Dec 19, 2024 · 969f993 · 969f993
1 parent d07a3f7
commit 969f993
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/.github/workflows/docker-build.yml b/.github/workflows/docker-build.yml
@@ -47,7 +47,7 @@ jobs:
           - { python_version: "3.11", pytorch_version: "2.2", cuda_version: "12.1.0" }
           - { python_version: "3.11", pytorch_version: "2.3", cuda_version: "12.1.0" }
           - { python_version: "3.11", pytorch_version: "2.4", cuda_version: "12.1.0" }
-          - { python_version: "3.12", pytorch_version: "2.5", cuda_version: "12.1.0", latest: true }
+          - { python_version: "3.12", pytorch_version: "2.5", cuda_version: "12.1.0", tag: "--add_latest" }
     steps:
       - uses: actions/checkout@v4
         with:
@@ -66,12 +66,11 @@ jobs:
       - name: Set tags
         run: |
           pip install -q -r .actions/requirements.txt
-          tags = $(python .actions/assistant.py generate_docker_tags \
-            --release-version="${{ env.RELEASE_VERSION }}" \
-            --python-version="${{ matrix.python_version }}" \
-            --torch-version="${{ matrix.pytorch_version }}" \
-            --cuda-version="${{ matrix.cuda_version }}" \
-            --add_latest="${{ matrix.latest }}")
+          tags = $(python .actions/assistant.py generate-docker-tags \
+            --release_version="${{ env.RELEASE_VERSION }}" \
+            --python_version="${{ matrix.python_version }}" \
+            --torch_version="${{ matrix.pytorch_version }}" \
+            --cuda_version="${{ matrix.cuda_version }}" ${{ matrix.tag }} )
           echo "DOCKER_TAGS=$tags" >> $GITHUB_ENV
 
       - uses: docker/build-push-action@v6