Merge pull request #379 from puneetmatharu/patch-use-of-dockerize-in-tf-build-sh

Patch failing TensorFlow build in CI

Author: nSircombe

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

ML-Frameworks/tensorflow-aarch64/get-source.sh

@@ -31,5 +31,5 @@ git-shallow-clone https://github.com/tensorflow/tensorflow.git $TENSORFLOW_HASH
     # Apply TensorFlow WIP patches here
 
     # https://github.com/tensorflow/tensorflow/pull/100882 - build(aarch64): Update Compute Library to 52.4.0
-    apply-github-patch tensorflow/tensorflow 0cd9bed2d5f5456576d3bcd0b80aba7cd478ee26
+    apply-github-patch tensorflow/tensorflow 33a28c399c24ed03e94bd8d8fee289f67946fc7b
 )