Update patch releases on 0.17 (#2723)

* Update Compatibility Matrix (#2709)

* Cleanup `tfa.image.equalize` (#2711)

* Update patch releases for breaking protobuf

Co-authored-by: Lukas Geiger <lukas.geiger94@gmail.com>

Author: seanpmorgan

.github/workflows/release.yml

@@ -42,19 +42,19 @@ jobs:
         # https://github.com/bazelbuild/bazel/issues/14232#issuecomment-1011247429
         os: ['macos-10.15', 'windows-2019', 'ubuntu-18.04']
         py-version: ['3.7', '3.8', '3.9', '3.10']
-        tf-version: ['2.7.2', '2.8.1', '2.9.0']
+        tf-version: ['2.7.3', '2.8.2', '2.9.1']
         cpu: ['x86']
         exclude:
           - py-version: '3.10'
-            tf-version: '2.7.2'
+            tf-version: '2.7.3'
         include:
           - os: 'macos-11'
             cpu: 'arm64'
-            tf-version: '2.9.0'
+            tf-version: '2.9.1'
             py-version: '3.8'
           - os: 'macos-11'
             cpu: 'arm64'
-            tf-version: '2.9.0'
+            tf-version: '2.9.1'
             py-version: '3.9'
       fail-fast: false
     steps:
@@ -97,16 +97,16 @@ jobs:
       matrix:
         os: ['macOS', 'Windows', 'Linux']
         py-version: ['3.7', '3.8', '3.9', '3.10']
-        tf-version: ['2.9.0']
+        tf-version: ['2.9.1']
         cpu: ['x86']
         include:
           - os: 'macOS'
             cpu: 'arm64'
-            tf-version: '2.9.0'
+            tf-version: '2.9.1'
             py-version: '3.8'
           - os: 'macOS'
             cpu: 'arm64'
-            tf-version: '2.9.0'
+            tf-version: '2.9.1'
             py-version: '3.9'
       fail-fast: false
     if: (github.event_name == 'push' && github.ref == 'refs/heads/master') || github.event_name == 'release'

CONTRIBUTING.md

@@ -174,7 +174,7 @@ Just run from the root:
 
 ```bash
 pip install tensorflow==2.8
-# you can use "pip install tensorflow-cpu==2.9.0" too if you're not testing on gpu.
+# you can use "pip install tensorflow-cpu==2.9.1" too if you're not testing on gpu.
 pip install -e ./
 ```
 

README.md

@@ -81,6 +81,7 @@ what it was tested against.
 | TensorFlow Addons | TensorFlow | Python  |
 |:----------------------- |:---|:---------- |
 | tfa-nightly | 2.7, 2.8, 2.9 | 3.7, 3.8, 3.9, 3.10 | 
+| tensorflow-addons-0.17.0 | 2.7, 2.8, 2.9 |3.7, 3.8, 3.9, 3.10 |
 | tensorflow-addons-0.16.1 | 2.6, 2.7, 2.8 |3.7, 3.8, 3.9, 3.10 |
 | tensorflow-addons-0.15.0 | 2.5, 2.6, 2.7 |3.7, 3.8, 3.9 |
 | tensorflow-addons-0.14.0 | 2.4, 2.5, 2.6 |3.6, 3.7, 3.8, 3.9 |
@@ -112,6 +113,7 @@ is compiled differently. A typical example of this would be `conda`-installed Te
 | TensorFlow Addons | TensorFlow | Compiler  | cuDNN | CUDA | 
 |:----------------------- |:---- |:---------|:---------|:---------|
 | tfa-nightly | 2.9 | GCC 7.3.1 | 8.1 | 11.2 |
+| tensorflow-addons-0.17.0 | 2.9  | GCC 7.3.1 | 8.1 | 11.2 |
 | tensorflow-addons-0.16.1 | 2.8  | GCC 7.3.1 | 8.1 | 11.2 |
 | tensorflow-addons-0.15.0 | 2.7  | GCC 7.3.1 | 8.1 | 11.2 |
 | tensorflow-addons-0.14.0 | 2.6  | GCC 7.3.1 | 8.1 | 11.2 |

tensorflow_addons/image/color_ops.py

@@ -79,8 +79,8 @@ def equalize(
     with tf.name_scope(name or "equalize"):
         image_dims = tf.rank(image)
         image = to_4D_image(image)
-        fn = partial(_equalize_image)
-        image = tf.map_fn(lambda x: fn(x, bins), image)
+        fn = partial(_equalize_image, bins=bins)
+        image = tf.map_fn(fn, image)
         return from_4D_image(image, image_dims)
 
 

tools/build_dev_container.sh

@@ -4,7 +4,7 @@ set -x -e
 
 docker build \
     -f tools/docker/dev_container.Dockerfile \
-    --build-arg TF_VERSION=2.9.0 \
+    --build-arg TF_VERSION=2.9.1 \
     --build-arg TF_PACKAGE=tensorflow \
     --build-arg PY_VERSION=$PY_VERSION \
     --no-cache \

tools/docker/cpu_tests.Dockerfile

@@ -1,7 +1,7 @@
 #syntax=docker/dockerfile:1.1.5-experimental
 FROM python:3.7 as build_wheel
 
-ARG TF_VERSION=2.9.0
+ARG TF_VERSION=2.9.1
 RUN pip install --default-timeout=1000 tensorflow-cpu==$TF_VERSION
 
 RUN apt-get update && apt-get install -y sudo rsync

tools/install_deps/tensorflow-cpu.txt

@@ -1 +1 @@
-tensorflow-cpu~=2.9.0
+tensorflow-cpu~=2.9.1

tools/install_deps/tensorflow.txt

@@ -1 +1 @@
-tensorflow~=2.9.0
+tensorflow~=2.9.1

tools/run_gpu_tests.sh

@@ -6,7 +6,7 @@ export DOCKER_BUILDKIT=1
 docker build \
        -f tools/docker/build_wheel.Dockerfile \
        --target tfa_gpu_tests \
-       --build-arg TF_VERSION=2.9.0 \
+       --build-arg TF_VERSION=2.9.1 \
        --build-arg PY_VERSION=3.7 \
        -t tfa_gpu_tests ./
 docker run --rm -t --gpus=all tfa_gpu_tests