fix bert can not download (#2944)

Co-authored-by: ardaci <dongjie.shi@intel.com>
intel-analytics · Oct 12, 2020 · 446a36a · 446a36a
1 parent b7ee3e1
commit 446a36a
Showing 1 changed file with 9 additions and 1 deletion.
diff --git 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=$?