Fix MBCC & scala publish pipeline

apache · Feb 21, 2020 · 74adc35 · 74adc35
1 parent 9dcf71d
commit 74adc35
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ci/publish/Jenkinsfile b/ci/publish/Jenkinsfile
@@ -36,7 +36,7 @@ utils.assign_node_labels(utility: 'restricted-utility', linux_cpu: 'restricted-m
 // CPU and GPU. OSX nodes are not currently supported by Jenkins
 def nodeMap = ['cpu': NODE_LINUX_CPU, 'gpu': NODE_LINUX_GPU_P3]
 def scalaOSMap = ['cpu': 'linux-x86_64-cpu', 'gpu': 'linux-x86_64-gpu']
-def scalaVariantMap = ['cpu': 'mkl', 'gpu': 'cu92mkl']
+def scalaVariantMap = ['cpu': 'cpu', 'gpu': 'cu92']
 
 def wrapStep(nodeToRun, workspaceName, step) {
   return {

diff --git a/tests/nightly/model_backwards_compatibility_check/train_mxnet_legacy_models.sh b/tests/nightly/model_backwards_compatibility_check/train_mxnet_legacy_models.sh
@@ -26,14 +26,14 @@ run_models() {
 	echo "Running training files and preparing models"
 	echo '=========================='
   export MXNET_ENFORCE_DETERMINISM=1
-	python model_backwards_compat_train.py
+	python3 model_backwards_compat_train.py
 	echo '=========================='
 }
 
 install_mxnet() {
 	version=$1
 	echo "Installing MXNet "$version
-	pip install mxnet==$version --user
+	pip3 install mxnet==$version --user
 }
 
 ## Cuts the string and gives only the major version part.