Update correct bazel version in build_aar_with_docker.sh

PiperOrigin-RevId: 352709339
Change-Id: I746a7ae6fc2e94f711050edeea6f0055fdddc5f4
diff --git a/tensorflow/lite/tools/build_aar_with_docker.sh b/tensorflow/lite/tools/build_aar_with_docker.sh
index 1bd3c07..02f1336 100755
--- a/tensorflow/lite/tools/build_aar_with_docker.sh
+++ b/tensorflow/lite/tools/build_aar_with_docker.sh
@@ -114,6 +114,10 @@
   git pull -a
   git checkout ${FLAG_CHECKPOINT}
 
+  # Configure Bazel.
+  source tensorflow/tools/ci_build/release/common.sh
+  install_bazelisk
+
   # Building with bazel.
   export BAZEL_CACHE_DIR=${BAZEL_CACHE_DIR}
   export OMIT_PRINTING_OUTPUT_PATHS=YES