[fix] CI fail for protobuf update.

- Also include the horovod compile fail on macOS(They was caused by the same reason)

Author: rhdong

.github/workflows/ci_test.yml

@@ -91,6 +91,7 @@ jobs:
       run: |
         # Run on all notebooks to prevent upstream change.
         echo "Check formatting with nbfmt:"
+        python3 -m pip install --upgrade protobuf==3.20.0
         python3 -m tensorflow_docs.tools.nbfmt --test \
             $(find docs/tutorials/ -type f -name *.ipynb)
   nblint:
@@ -105,6 +106,7 @@ jobs:
       run: |
         # Run on all notebooks to prevent upstream change.
         echo "Lint check with nblint:"
+        python3 -m pip install --upgrade protobuf==3.20.0
         python3 -m tensorflow_docs.tools.nblint \
             --arg=repo:tensorflow/recommenders-addons \
             --exclude_lint=tensorflow::button_colab \

.github/workflows/make_wheel_Windows_x86.sh

@@ -10,6 +10,8 @@ fi
 python -m pip install --default-timeout=1000 wheel setuptools tensorflow==$TF_VERSION horovod==$HOROVOD_VERSION
 bash ./tools/testing/build_and_run_tests.sh
 
+python -m pip install --upgrade protobuf==3.20.0
+
 python configure.py
 
 bazel.exe build  --no-cache \

.github/workflows/make_wheel_macOS_arm64.sh

@@ -11,6 +11,7 @@ export TF_NEED_CUDA=0
 
 python --version
 
+RUN python -m pip install --upgrade protobuf==3.20.0
 python configure.py
 
 bazel build \

.github/workflows/make_wheel_macOS_x86.sh

@@ -9,6 +9,8 @@ python --version
 brew install open-mpi
 
 python -m pip install --default-timeout=1000 delocate==0.9.1 wheel setuptools tensorflow==$TF_VERSION
+python -m pip install --upgrade protobuf==3.20.0
+
 bash tools/docker/install/install_horovod.sh $HOROVOD_VERSION --only-cpu
 
 bash tools/testing/build_and_run_tests.sh

.github/workflows/release.yml

@@ -34,6 +34,7 @@ jobs:
           pip install --default-timeout=1000 -r tools/install_deps/pytest.txt -r tools/install_deps/tensorflow-cpu.txt -r requirements.txt
           sudo apt install -y redis > /dev/null 2> /dev/null
           bash tools/install_deps/install_bazelisk.sh ./
+          python -m pip install --upgrade protobuf==3.20.0
           python configure.py
           bazel test -c opt -k --test_timeout 300,450,1200,3600 --test_output=errors //tensorflow_recommenders_addons/...
   release-wheel:

tools/docker/build_wheel.Dockerfile

@@ -43,6 +43,8 @@ RUN python -m pip install -r /install_deps/pytest.txt
 COPY requirements.txt .
 RUN python -m pip install -r requirements.txt
 
+RUN python -m pip install --upgrade protobuf==3.20.0
+
 COPY ./ /recommenders-addons
 WORKDIR /recommenders-addons
 
@@ -86,6 +88,8 @@ ARG TF_VERSION
 ARG TF_NAME
 RUN python -m pip install --default-timeout=1000 $TF_NAME==$TF_VERSION
 
+RUN python -m pip install --upgrade protobuf==3.20.0
+
 COPY --from=make_wheel /recommenders-addons/wheelhouse/ /recommenders-addons/wheelhouse/
 RUN pip install /recommenders-addons/wheelhouse/*.whl
 

tools/docker/cpu_tests.Dockerfile

@@ -8,6 +8,8 @@ ARG HOROVOD_VERSION="0.23.0"
 
 RUN pip install --default-timeout=1000 tensorflow-cpu==$TF_VERSION
 
+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/
@@ -24,6 +26,9 @@ RUN pip install -r pytest.txt pytest-cov
 
 COPY ./ /recommenders-addons
 WORKDIR recommenders-addons
+
+RUN python -m pip install --upgrade protobuf==3.20.0
+
 RUN python configure.py
 RUN pip install -e ./
 RUN --mount=type=cache,id=cache_bazel,target=/root/.cache/bazel \
@@ -42,6 +47,8 @@ RUN pip install --default-timeout=1000 --upgrade --force-reinstall -r tensorflow
 
 COPY --from=0 /recommenders-addons/artifacts /artifacts
 
+RUN python -m pip install --upgrade protobuf==3.20.0
+
 RUN pip install /artifacts/tensorflow_recommenders_addons-*.whl
 
 # check that we didnd't forget to add a py file to

tools/docker/sanity_check.Dockerfile

@@ -6,6 +6,8 @@ COPY tools/install_deps/yapf.txt ./
 RUN pip install -r yapf.txt
 COPY ./ /recommenders-addons
 WORKDIR /recommenders-addons
+
+RUN python -m pip install --upgrade protobuf==3.20.0
 RUN python tools/check_python_format.py
 RUN touch /ok.txt
 
@@ -30,6 +32,9 @@ COPY ./ /recommenders-addons
 RUN pip install -e /recommenders-addons
 
 WORKDIR /recommenders-addons
+
+RUN python -m pip install --upgrade protobuf==3.20.0
+
 RUN python configure.py
 RUN --mount=type=cache,id=cache_bazel,target=/root/.cache/bazel \
     bash tools/install_so_files.sh
@@ -51,6 +56,9 @@ RUN ./install_bazel.sh $USE_BAZEL_VERSION
 
 COPY ./ /recommenders-addons
 WORKDIR /recommenders-addons
+
+RUN python -m pip install --upgrade protobuf==3.20.0
+
 RUN python ./configure.py
 RUN --mount=type=cache,id=cache_bazel,target=/root/.cache/bazel \
     bazel build --nobuild -- //tensorflow_recommenders_addons/...
@@ -104,6 +112,8 @@ RUN ./install_bazel.sh $USE_BAZEL_VERSION
 COPY ./ /recommenders-addons
 WORKDIR /recommenders-addons
 
+RUN python -m pip install --upgrade protobuf==3.20.0
+
 RUN python configure.py
 RUN --mount=type=cache,id=cache_bazel,target=/root/.cache/bazel \
     bash tools/install_so_files.sh
@@ -131,6 +141,9 @@ RUN ./install_bazel.sh $USE_BAZEL_VERSION
 
 COPY ./ /recommenders-addons
 WORKDIR /recommenders-addons
+
+RUN python -m pip install --upgrade protobuf==3.20.0
+
 RUN python configure.py
 RUN --mount=type=cache,id=cache_bazel,target=/root/.cache/bazel \
     bash tools/install_so_files.sh