From b07a7ff36614dcc82604b68f4786db38fdee6162 Mon Sep 17 00:00:00 2001
From: Aleksandr Laptev <alaptev@nvidia.com>
Date: Mon, 30 May 2022 13:09:51 -0700
Subject: [PATCH] verbose k2 install, skip if failed

Signed-off-by: Aleksandr Laptev <alaptev@nvidia.com>
---
 Dockerfile                             | 2 +-
 scripts/speech_recognition/k2/setup.sh | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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!"