fix env name

yfukai · Jan 29, 2024 · e0f9fa4 · e0f9fa4
1 parent 3571fea
commit e0f9fa4
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/.github/workflows/docker-publish.yml b/.github/workflows/docker-publish.yml
@@ -71,9 +71,6 @@ jobs:
           images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
 
 
-      - name: Get the tag name
-        run: echo "TAG=${GITHUB_REF/refs\/tags\//}" >> $GITHUB_ENV
-
       - name: Set up QEMU
         uses: docker/setup-qemu-action@v2
 
@@ -89,7 +86,7 @@ jobs:
           tags: ${{ steps.meta_jax_conda.outputs.tags }}
           labels: ${{ steps.meta_jax_conda.outputs.labels }}
           build-args: |
-            JAX_VERSION=${{ env.JAX_version }}
+            JAX_VERSION=${{ env.JAX_VERSION }}
 
       - name: Build and push Docker image (JAX conda GPU)
         id: build-and-push-jax-conda-gpu
@@ -101,7 +98,7 @@ jobs:
           tags: ${{ steps.meta_jax_conda.outputs.tags }}-cuda
           labels: ${{ steps.meta_jax_conda.outputs.labels }}
           build-args: |
-            JAX_VERSION=${{ env.JAX_version }}
+            JAX_VERSION=${{ env.JAX_VERSION }}
             JAX_VERSION_EXTRA="cuda11_local"
             BASE_IMAGE="nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04"