Use bazelisk (tensorflow#2097)

* Use bazelisk to install bazel

.bazelversion

@@ -0,0 +1 @@
+3.1.0

.github/workflows/ci_test.yml

@@ -10,9 +10,6 @@ on:
       - master
       - r*
 
-env:
-  BAZEL_VERSION: 3.1.0
-
 jobs:
   flake8-test:
     name: Flake8

.github/workflows/release.yml

@@ -15,7 +15,6 @@ on:
       - r*
 
 env:
-  BAZEL_VERSION: '3.1.0'
   MIN_PY_VERSION: '3.5'
   MAX_PY_VERSION: '3.8'
 
@@ -31,7 +30,7 @@ jobs:
       - name: Build wheels
         run: |
           pip install --default-timeout=1000 -r tools/install_deps/pytest.txt -r tools/install_deps/tensorflow-cpu.txt -r requirements.txt
-          bash tools/install_deps/bazel_linux.sh
+          bash tools/install_deps/install_bazelisk.sh ./
           python configure.py
           bazel test -c opt -k --test_timeout 300,450,1200,3600 --test_output=errors //tensorflow_addons/...
   release-wheel:
@@ -60,11 +59,7 @@ jobs:
       - name: Setup Bazel
         # Ubuntu bazel is run inside of the docker image
         if: matrix.os != 'ubuntu-18.04'
-        # MacOS VMs share IP addr on GH actions. This bazel installation authenticates token to prevent api rate limiting.
-        uses: jwlawson/actions-setup-bazel@784af34 # Version 1.1
-        with:
-          bazel-version: ${{ env.BAZEL_VERSION}}
-          github-api-token: ${{ secrets.GITHUB_TOKEN }}
+        run: bash tools/install_deps/install_bazelisk.sh ./
       - name: Build wheels
         env:
           OS: ${{ runner.os }}

tools/docker/cpu_tests.Dockerfile

@@ -5,8 +5,8 @@ ARG TF_VERSION=2.3.0
 RUN pip install --default-timeout=1000 tensorflow-cpu==$TF_VERSION
 
 RUN apt-get update && apt-get install -y sudo rsync
-COPY tools/install_deps/bazel_linux.sh ./
-RUN bash bazel_linux.sh
+COPY tools/install_deps/install_bazelisk.sh .bazelversion ./
+RUN bash install_bazelisk.sh
 
 COPY requirements.txt ./
 RUN pip install -r requirements.txt

tools/docker/sanity_check.Dockerfile

@@ -40,8 +40,8 @@ COPY tools/install_deps/tensorflow-cpu.txt ./
 RUN pip install --default-timeout=1000 -r tensorflow-cpu.txt
 
 RUN apt-get update && apt-get install sudo
-COPY tools/install_deps/bazel_linux.sh ./
-RUN bash bazel_linux.sh
+COPY tools/install_deps/install_bazelisk.sh .bazelversion ./
+RUN bash install_bazelisk.sh
 
 COPY ./ /addons
 WORKDIR /addons
@@ -109,8 +109,8 @@ COPY tools/install_deps/pytest.txt ./
 RUN pip install -r pytest.txt
 
 RUN apt-get update && apt-get install -y sudo rsync
-COPY tools/install_deps/bazel_linux.sh ./
-RUN bash bazel_linux.sh
+COPY tools/install_deps/install_bazelisk.sh .bazelversion ./
+RUN bash install_bazelisk.sh
 
 COPY ./ /addons
 WORKDIR /addons

tools/install_deps/install_bazelisk.sh

@@ -0,0 +1,23 @@
+# Downloads bazelisk to ${output_dir} as `bazel`.
+date
+
+output_dir=${1:-"/usr/local/bin"}
+
+case "$(uname -s)" in
+    Darwin) name=bazelisk-darwin-amd64 ;;
+    Linux)  name=bazelisk-linux-amd64  ;;
+    *) name=bazelisk-windows-amd64 ;;
+esac
+
+mkdir -p "${output_dir}"
+curl -LO "https://github.com/bazelbuild/bazelisk/releases/download/v1.3.0/${name}"
+
+mv "${name}" "${output_dir}/bazel"
+chmod u+x "${output_dir}/bazel"
+
+if [[ ! ":$PATH:" =~ :${output_dir}/?: ]]; then
+    PATH="${output_dir}:$PATH"
+fi
+
+which bazel
+date