From 12b41b2657e44d25606caabb03e4b5dc0bf36635 Mon Sep 17 00:00:00 2001
From: Mathieu Broillet <mathieu@broillet.ch>
Date: Wed, 28 Aug 2024 15:28:23 +0200
Subject: [PATCH] try to fix env to workflow

---
 .github/workflows/build-and_publish-prebuilts-whl.yml | 7 ++-----
 prebuilts/Dockerfile                                  | 5 -----
 2 files changed, 2 insertions(+), 10 deletions(-)

diff --git a/.github/workflows/build-and_publish-prebuilts-whl.yml b/.github/workflows/build-and_publish-prebuilts-whl.yml
index 9848bee..d86379b 100644
--- a/.github/workflows/build-and_publish-prebuilts-whl.yml
+++ b/.github/workflows/build-and_publish-prebuilts-whl.yml
@@ -30,13 +30,10 @@ jobs:
 
       # Step 3: Build the Docker image
       - name: Build Docker image
-        env:
-          ROCM_VERSION: ${{ inputs.rocm_version }}
-          TORCH_VERSION: ${{ inputs.torch_version }}
         run: |
           docker build \
-            --build-arg ROCM_VERSION=${ROCM_VERSION} \
-            --build-arg TORCH_VERSION=${TORCH_VERSION} \
+            --build-arg ROCM_VERSION=${{ inputs.rocm_version }} \
+            --build-arg TORCH_VERSION=${{ inputs.torch_version }} \
             -t prebuilt-wheels ./prebuilts
 
       # Step 4: Create a container and run the script
diff --git a/prebuilts/Dockerfile b/prebuilts/Dockerfile
index 2829975..7533cd7 100644
--- a/prebuilts/Dockerfile
+++ b/prebuilts/Dockerfile
@@ -1,8 +1,3 @@
-ARG ROCM_VERSION
-ARG TORCH_VERSION
-ENV ROCM_VERSION=${ROCM_VERSION}
-ENV TORCH_VERSION=${TORCH_VERSION}
-
 FROM rocm/dev-ubuntu-22.04:${ROCM_VERSION}
 
 ENV DEBIAN_FRONTEND=noninteractive \