From 446a36a0040e1df07d55de1f1b47079e15321f82 Mon Sep 17 00:00:00 2001
From: pinggao187 <44044110+pinggao187@users.noreply.github.com>
Date: Mon, 12 Oct 2020 15:53:32 +0800
Subject: [PATCH] fix bert can not download (#2944)

Co-authored-by: ardaci <dongjie.shi@intel.com>
---
 scripts/get_python_packages | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/scripts/get_python_packages b/scripts/get_python_packages
index 0716cc1366c..968ba3516f1 100755
--- a/scripts/get_python_packages
+++ b/scripts/get_python_packages
@@ -67,7 +67,15 @@ then
 fi
 
 echo "Fetching bert with ${BERT_COMMIT_ID}"
-GET_COMMAND="wget -nc $FTP_URI/bert/archive/${BERT_COMMIT_ID}.zip -P $DEST_DIR"
+if [ $FTP_URI ];
+then
+    PREFIX="$FTP_URI/bert"
+else
+    PREFIX=https://github.com/google-research/bert
+
+fi
+
+GET_COMMAND="wget -nc $PREFIX/archive/${BERT_COMMIT_ID}.zip -P $DEST_DIR"
 echo "Running: $GET_COMMAND"
 $GET_COMMAND
 exit_status=$?