Adding a way to pass extra index urls to s2i python wrappers

SeldonIO · Dec 11, 2019 · 038eba6 · 038eba6
1 parent 5bf3894
commit 038eba6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/wrappers/s2i/python/s2i/bin/assemble b/wrappers/s2i/python/s2i/bin/assemble
@@ -51,6 +51,12 @@ echo "---> Installing application source..."
 cp -Rf /tmp/src/. ./
 
 if [[ -f requirements.txt ]]; then
+
+  if [[ -n "$EXTRA_INDEX_URL" ]]; then
+     echo "---> Adding extra index url to pip..."
+     pip config set global.extra-index-url "${EXTRA_INDEX_URL}"
+  fi
+
   echo "---> Installing dependencies ..."
   pip install --find-links /whl -r requirements.txt
 elif [[ -f setup.py ]]; then