Patch failing TensorFlow CI.

Author: puneetmatharu

ML-Frameworks/tensorflow-aarch64/build.sh

@@ -72,8 +72,4 @@ fi
 tf_wheel_name=$(grep -o "tensorflow-.*.whl" $build_log | head -n -1 | tail -n 1)
 echo $tf_wheel_name
 
-docker build -t toolsolutions-tensorflow:latest \
-    --build-arg TENSORFLOW_WHEEL=results/$tf_wheel_name \
-    --build-arg DOCKER_IMAGE_MIRROR \
-    --build-arg USERNAME=ubuntu \
-    .
+./dockerize.sh "results/$tf_wheel_name" --build-only

ML-Frameworks/tensorflow-aarch64/dockerize.sh

@@ -21,7 +21,7 @@ set -eux -o pipefail
 
 help_str="dockerize.sh takes a TensorFlow wheel as argument. It \
 installs the wheel inside a Docker container with examples and requirements."
-if [ "$#" -ne 1 ]; then
+if [ "$#" -lt 1 ]; then
     echo $help_str
     exit 1
 fi