Finish the bazel install before building. (#1135)

Author: gabrieldemarmiesse

tools/docker/cpu_tests.Dockerfile

@@ -10,6 +10,9 @@ RUN bash bazel.sh
 COPY requirements.txt ./
 RUN pip install -r requirements.txt
 
+COPY tools/docker/finish_bazel_install.sh ./
+RUN bash finish_bazel_install.sh
+
 COPY ./ /addons
 WORKDIR addons
 RUN bash tools/ci_testing/addons_cpu.sh --no-deps

tools/docker/finish_bazel_install.sh

@@ -0,0 +1,12 @@
+set -e
+# hack to have the dependecies in the docker cache
+# and have much faster local build with docker
+# can be removed once docker buildx/buildkit is stable
+# and we'll use "RUN --mount=cache ..." instead
+cd /tmp
+git clone https://github.com/tensorflow/addons.git
+cd addons
+python ./configure.py --no-deps
+bazel build --nobuild -- //tensorflow_addons/...
+cd ..
+rm -rf ./addons

tools/docker/gpu_tests.Dockerfile

@@ -7,6 +7,10 @@ RUN python3 -m pip install -r build-requirements.txt
 COPY requirements.txt ./
 RUN python3 -m pip install -r requirements.txt
 
+COPY tools/docker/finish_bazel_install.sh ./
+RUN bash finish_bazel_install.sh
+
+
 COPY ./ /addons
 WORKDIR addons
 CMD ["bash", "tools/ci_testing/addons_gpu.sh"]

tools/docker/sanity_check.Dockerfile

@@ -45,6 +45,9 @@ RUN apt-get update && apt-get install sudo
 COPY tools/ci_build/install/bazel.sh ./
 RUN bash bazel.sh
 
+COPY tools/docker/finish_bazel_install.sh ./
+RUN bash finish_bazel_install.sh
+
 COPY ./ /addons
 WORKDIR /addons
 RUN python ./configure.py --no-deps