Add .bazelversion file

.bazelversion

@@ -0,0 +1 @@
+5.1.1

.github/workflows/ci_test.yml

@@ -9,8 +9,6 @@ on:
     branches:
       - master
       - r*
-env:
-  USE_BAZEL_VERSION: "5.1.1"
 
 jobs:
   yapf-test:

.github/workflows/release.yml

@@ -17,7 +17,6 @@ on:
 env:
   MIN_PY_VERSION: '3.7'
   MAX_PY_VERSION: '3.9'
-  USE_BAZEL_VERSION: "5.1.1"
   HOROVOD_VERSION: '0.23.0'
 
 jobs:

tools/docker/cpu_tests.Dockerfile

@@ -2,7 +2,6 @@
 FROM python:3.7 as build_wheel
 
 ARG TF_VERSION=2.8.3
-ARG USE_BAZEL_VERSION=5.1.1
 ARG MPI_VERSION="4.1.1"
 ARG HOROVOD_VERSION="0.23.0"
 
@@ -12,8 +11,8 @@ RUN python -m pip install --upgrade protobuf==3.20.0
 
 RUN apt-get update && apt-get install -y sudo rsync cmake openmpi-bin libopenmpi-dev
 
-COPY tools/docker/install/install_bazel.sh /install/
-RUN  /install/install_bazel.sh $USE_BAZEL_VERSION
+COPY tools/install_deps/install_bazelisk.sh /install/
+RUN bash /install/install_bazelisk.sh
 
 COPY tools/docker/install/install_horovod.sh /install/
 RUN  /install/install_horovod.sh $HOROVOD_VERSION --only-cpu

tools/docker/sanity_check.Dockerfile

@@ -14,8 +14,6 @@ RUN touch /ok.txt
 # -------------------------------
 FROM python:3.7 as source_code_test
 
-ARG USE_BAZEL_VERSION
-
 COPY tools/install_deps /install_deps
 RUN --mount=type=cache,id=cache_pip,target=/root/.cache/pip \
     cd /install_deps && pip install \
@@ -25,8 +23,8 @@ RUN --mount=type=cache,id=cache_pip,target=/root/.cache/pip \
     -r pytest.txt
 
 RUN apt-get update && apt-get install -y sudo rsync cmake
-COPY tools/docker/install/install_bazel.sh ./
-RUN ./install_bazel.sh $USE_BAZEL_VERSION
+COPY tools/install_deps/install_bazelisk.sh .bazelversion ./
+RUN bash install_bazelisk.sh
 
 COPY ./ /recommenders-addons
 RUN pip install -e /recommenders-addons
@@ -47,14 +45,12 @@ RUN touch /ok.txt
 # -------------------------------
 FROM python:3.7 as valid_build_files
 
-ARG USE_BAZEL_VERSION
-
 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/docker/install/install_bazel.sh ./
-RUN ./install_bazel.sh $USE_BAZEL_VERSION
+COPY tools/install_deps/install_bazelisk.sh .bazelversion ./
+RUN bash install_bazelisk.sh
 
 COPY ./ /recommenders-addons
 WORKDIR /recommenders-addons
@@ -98,8 +94,6 @@ RUN touch /ok.txt
 # docs tests
 FROM python:3.7 as docs_tests
 
-ARG USE_BAZEL_VERSION
-
 COPY tools/install_deps/tensorflow-cpu.txt ./
 RUN pip install --default-timeout=1000 -r tensorflow-cpu.txt
 COPY requirements.txt ./
@@ -109,8 +103,8 @@ COPY tools/install_deps/doc_requirements.txt ./
 RUN pip install -r doc_requirements.txt
 
 RUN apt-get update && apt-get install -y sudo rsync cmake
-COPY tools/docker/install/install_bazel.sh ./
-RUN ./install_bazel.sh $USE_BAZEL_VERSION
+COPY tools/install_deps/install_bazelisk.sh .bazelversion ./
+RUN bash install_bazelisk.sh
 
 COPY ./ /recommenders-addons
 WORKDIR /recommenders-addons
@@ -129,8 +123,6 @@ RUN touch /ok.txt
 # test the editable mode
 FROM python:3.7 as test_editable_mode
 
-ARG USE_BAZEL_VERSION
-
 COPY tools/install_deps/tensorflow-cpu.txt ./
 RUN pip install --default-timeout=1000 -r tensorflow-cpu.txt
 COPY requirements.txt ./
@@ -139,8 +131,8 @@ COPY tools/install_deps/pytest.txt ./
 RUN pip install -r pytest.txt
 
 RUN apt-get update && apt-get install -y sudo rsync cmake
-COPY tools/docker/install/install_bazel.sh ./
-RUN ./install_bazel.sh $USE_BAZEL_VERSION
+COPY tools/install_deps/install_bazelisk.sh .bazelversion ./
+RUN bash install_bazelisk.sh
 
 COPY ./ /recommenders-addons
 WORKDIR /recommenders-addons

tools/install_deps/install_bazelisk.sh

@@ -1,18 +1,14 @@
+#!/usr/bin/env bash
+set -x -e
+
 # 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}"
+wget --progress=dot:mega -O ${output_dir}/bazel https://github.com/bazelbuild/bazelisk/releases/latest/download/bazelisk-linux-$([ $(uname -m) = "aarch64" ] && echo "arm64" || echo "amd64")
 
-mv "${name}" "${output_dir}/bazel"
 chmod u+x "${output_dir}/bazel"
 
 if [[ ! ":$PATH:" =~ :${output_dir}/?: ]]; then

tools/run_build.sh

@@ -6,5 +6,4 @@ set -e
 
 DOCKER_BUILDKIT=1 docker build \
     -f tools/docker/sanity_check.Dockerfile \
-    --build-arg USE_BAZEL_VERSION=${USE_BAZEL_VERSION:-"5.1.1"} \
     --target=${1} ./