From 377c5340d084a58fccfda59621bcb8491398a8e9 Mon Sep 17 00:00:00 2001
From: Raphael Glon <oOraph@users.noreply.github.com>
Date: Tue, 9 Apr 2024 10:36:22 +0200
Subject: [PATCH] Misc fixes

Signed-off-by: Raphael Glon <oOraph@users.noreply.github.com>
---
 .github/workflows/test_inf2_tgi.yml  |  2 --
 text-generation-inference/Dockerfile | 10 ++--------
 2 files changed, 2 insertions(+), 10 deletions(-)

diff --git a/.github/workflows/test_inf2_tgi.yml b/.github/workflows/test_inf2_tgi.yml
index 0e8739b77..1b25fe344 100644
--- 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
\ No newline at end of file
diff --git a/text-generation-inference/Dockerfile b/text-generation-inference/Dockerfile
index 1066af184..299ba3ec0 100644
--- 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 \