From 0a215304b233742c36a426bfe4646c86921b4bac Mon Sep 17 00:00:00 2001
From: Justin <justinedwards1230@gmail.com>
Date: Mon, 21 Oct 2024 10:20:47 -0400
Subject: [PATCH] fix: env vars in workflow

---
 .github/workflows/docker-build.yaml | 21 ++++++++-------------
 1 file changed, 8 insertions(+), 13 deletions(-)

diff --git a/.github/workflows/docker-build.yaml b/.github/workflows/docker-build.yaml
index f376f14..5dab1b5 100644
--- a/.github/workflows/docker-build.yaml
+++ b/.github/workflows/docker-build.yaml
@@ -7,25 +7,20 @@ on:
             - main
             - dev
 
-env:
-    CUDA_IMAGE_TAG: type=raw,enable=${{ github.ref == 'refs/heads/main' }},prefix=,suffix=,value=cuda
-    CUDA_IMAGE_FLAVOR: suffix=-cuda,onlatest=true
-    IMAGE_NAME: ${{ github.repository }}
-
 jobs:
     build-main-image:
         uses: ./.github/workflows/build-docker-image.yaml
         with:
-            image_name: ${{ env.IMAGE_NAME }}
+            image_name: ${{ github.repository }}
             cache_id: main
 
     build-cuda-image:
         uses: ./.github/workflows/build-docker-image.yaml
         with:
-            image_name: ${{ env.IMAGE_NAME }}
+            image_name: ${{ github.repository }}
             cache_id: cuda
-            image_tag: ${{ env.CUDA_IMAGE_TAG }}
-            extract_flavor: ${{ env.CUDA_IMAGE_FLAVOR }}
+            image_tag: type=raw,enable=${{ github.ref == 'refs/heads/main' }},prefix=,suffix=,value=cuda
+            extract_flavor: suffix=-cuda,onlatest=true
             build_args: |
                 USE_CUDA=true
 
@@ -33,14 +28,14 @@ jobs:
         uses: ./.github/workflows/merge-docker-images.yaml
         needs: [build-main-image]
         with:
-            image_name: ${{ env.IMAGE_NAME }}
+            image_name: ${{ github.repository }}
             cache_id: main
 
     merge-cuda-images:
         uses: ./.github/workflows/merge-docker-images.yaml
         needs: [build-cuda-image]
         with:
-            image_name: ${{ env.IMAGE_NAME }}
+            image_name: ${{ github.repository }}
             cache_id: cuda
-            extract_flavor: ${{ env.CUDA_IMAGE_FLAVOR }}
-            extract_tags: ${{ env.CUDA_IMAGE_TAG }}
+            extract_flavor: suffix=-cuda,onlatest=true
+            extract_tags: type=raw,enable=${{ github.ref == 'refs/heads/main' }},prefix=,suffix=,value=cuda