diff --git a/Dockerfile b/Dockerfile index 07022b929710..29c92b6021a2 100644 --- a/Dockerfile +++ b/Dockerfile @@ -58,7 +58,7 @@ RUN for f in $(ls requirements*.txt); do pip install --disable-pip-version-check # install k2, skip if installation fails COPY scripts /tmp/nemo/scripts/ -RUN /bin/bash /tmp/nemo/scripts/speech_recognition/k2/setup.sh; exit 0 +RUN /bin/bash /tmp/nemo/scripts/speech_recognition/k2/setup.sh || exit 0 # copy nemo source into a scratch image FROM scratch as nemo-src diff --git a/scripts/speech_recognition/k2/setup.sh b/scripts/speech_recognition/k2/setup.sh index e3d1c475d23e..e110e99c3088 100755 --- a/scripts/speech_recognition/k2/setup.sh +++ b/scripts/speech_recognition/k2/setup.sh @@ -20,5 +20,5 @@ LATEST_RELEASE=$(git -c 'versionsort.suffix=-' \ | tail --lines=1 \ | cut --delimiter='/' --fields=3) -K2_MAKE_ARGS="-j" pip install git+${K2_REPO}@${LATEST_RELEASE}#egg=k2 || (echo "k2 could not be installed!"; exit 1) +K2_MAKE_ARGS="-j" pip install -v git+${K2_REPO}@${LATEST_RELEASE}#egg=k2 || (echo "k2 could not be installed!"; exit 1) python3 -m k2.version > /dev/null || (echo "k2 installed with errors! Please check installation manually."; exit 1) && echo "k2 installed successfully!"