experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix nightly smoke test after CL/301880779

Browse Source 
PiperOrigin-RevId: 302264423
Change-Id: If874506900e29cac7cbc457e7cc5e6a1cc0f5852
This commit is contained in:
Christian Sigg 2020-03-22 00:01:18 -07:00 committed by TensorFlower Gardener
parent 40675b3af0
commit b65dae5e5f
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/tools/ci_build/builds/nightly_release_smoke_test.sh
View File

@ -54,13 +54,15 @@ function test_tf_imports() {
# test for basic import and perform tf.add operation.
RET_VAL=$(python -c "import tensorflow as tf; t1=tf.constant([1,2,3,4]); t2=tf.constant([5,6,7,8]); print(tf.add(t1,t2).shape)")
if ! [[ ${RET_VAL} == *'(4,)'* ]]; then
echo "Unexpected return value: ${RET_VALUE}"
echo "PIP test on virtualenv FAILED, will not upload ${WHL_NAME} package."
return 1
fi
# test basic keras is available
RET_VAL=$(python -c "import tensorflow as tf; print(tf.keras.__name__)")
if ! [[ ${RET_VAL} == *'tensorflow.python.keras.api._v2.keras'* ]]; then
if ! [[ ${RET_VAL} == *'tensorflow.keras'* ]]; then
echo "Unexpected return value: ${RET_VALUE}"
echo "PIP test on virtualenv FAILED, will not upload ${WHL_NAME} package."
return 1
fi
@ -68,6 +70,7 @@ function test_tf_imports() {
# similar test for estimator
RET_VAL=$(python -c "import tensorflow as tf; print(tf.estimator.__name__)")
if ! [[ ${RET_VAL} == *'tensorflow_estimator.python.estimator.api._v2.estimator'* ]]; then
echo "Unexpected return value: ${RET_VALUE}"
echo "PIP test on virtualenv FAILED, will not upload ${WHL_NAME} package."
return 1
fi