Misc fixes

Signed-off-by: Raphael Glon <[email protected]>
huggingface · Apr 9, 2024 · 377c534 · 377c534
1 parent f7e8d12
commit 377c534
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 10 deletions.
diff --git a/.github/workflows/test_inf2_tgi.yml b/.github/workflows/test_inf2_tgi.yml
@@ -53,7 +53,5 @@ jobs:
       - name: Run integration tests
         shell: bash
         run: |
-          # gawk is required when invoking the Makefile targets
-          sudo apt install gawk -y
           source aws_neuron_venv_pytorch/bin/activate
           HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_NEURON_CI }} make tgi_docker_test
diff --git a/text-generation-inference/Dockerfile b/text-generation-inference/Dockerfile
@@ -53,17 +53,12 @@ RUN apt-get update -y \
     && apt-get clean
 RUN pip3 --no-cache-dir install --upgrade pip
 
-# VERSION is a mandatory parameter
-ARG VERSION
-RUN test -n ${VERSION:?}
-
 # Python server build image
 FROM base AS pyserver
 
 ARG VERSION
-RUN test -n ${VERSION:?}
 
-RUN apt-get update -y \
+RUN test -n ${VERSION:?} && apt-get update -y \
  && apt-get install -y --no-install-recommends \
     make \
     python3-venv \
@@ -81,10 +76,9 @@ RUN VERBOSE=1 BUILDDIR=/pyserver/build PROTODIR=/pyserver/proto VERSION=${VERSIO
 FROM base AS neuron
 
 ARG VERSION
-RUN test -n ${VERSION:?}
 
 # Install system prerequisites
-RUN apt-get update -y \
+RUN test -n ${VERSION:?} && apt-get update -y \
  && apt-get install -y --no-install-recommends \
     gnupg2 \
     wget \