message: Add a check for nightly binaries to prevent the size from exceeding 550MB.
diff: set -e
  set -x
  +MAX_WHL_SIZE=550M
  +
  function run_smoke_test() {
  VENV_TMP_DIR=$(mktemp -d)
  @@ -35,6 +37,9 @@ function run_smoke_test() {
  # Test TensorflowFlow imports
  test_tf_imports
  + # Test TensorFlow whl file size
  + test_tf_whl_size
  +
  RESULT=$?
  # Deactivate from virtualenv.
  deactivate || source deactivate || die "FAILED: Unable to deactivate from existing virtualenv."
  @@ -73,6 +78,14 @@ function test_tf_imports() {
  return $RESULT
  }
  +function test_tf_whl_size() {
  + if [[ $(find $WHL_NAME -type f -size +${MAX_WHL_SIZE}) ]]; then
  + echo "The whl size has exceeded 550MB. To keep within pypi's CDN
  +distribution limit, we must not exceed that threshold."
  + return 1
  + fi
  +}
  +
  ###########################################################################
  # Main
  ###########################################################################
  