Drop support for TF 2.1

.github/workflows/release.yml

@@ -41,7 +41,7 @@ jobs:
       matrix:
         os: ['macos-latest', 'windows-latest', 'ubuntu-18.04']
         py-version: ['3.5', '3.6', '3.7', '3.8']
-        tf-version: ['2.1.0', '2.2.0rc3']
+        tf-version: ['2.2.0rc3']
       fail-fast: false
     steps:
       - uses: actions/[email protected]
@@ -58,9 +58,7 @@ jobs:
         with:
           python-version: ${{ matrix.py-version }}
       - name: Build wheels
-        if: |
-          (matrix.py-version != '3.8' || matrix.tf-version != '2.1.0')
-          && (github.event_name != 'pull_request' || matrix.py-version == env.MIN_PY_VERSION || matrix.py-version == env.MAX_PY_VERSION)
+        if: github.event_name != 'pull_request' || matrix.py-version == env.MIN_PY_VERSION || matrix.py-version == env.MAX_PY_VERSION
         env:
           OS: ${{ runner.os }}
           PY_VERSION: ${{ matrix.py-version }}
@@ -69,9 +67,7 @@ jobs:
         shell: bash
         run: bash .github/workflows/make_wheel_${OS}.sh
       - uses: actions/upload-artifact@v1
-        if: |
-          (matrix.py-version != '3.8' || matrix.tf-version != '2.1.0')
-          && (github.event_name != 'pull_request' || matrix.py-version == env.MIN_PY_VERSION || matrix.py-version == env.MAX_PY_VERSION)
+        if: github.event_name != 'pull_request' || matrix.py-version == env.MIN_PY_VERSION || matrix.py-version == env.MAX_PY_VERSION
         with:
           name: ${{ runner.os }}-${{ matrix.py-version }}-tf${{ matrix.tf-version }}-wheel
           path: wheelhouse
@@ -88,7 +84,7 @@ jobs:
       - uses: actions/download-artifact@v1
         if: github.event_name != 'pull_request' || matrix.py-version == env.MIN_PY_VERSION  || matrix.py-version == env.MAX_PY_VERSION
         with:
-          name: ${{ matrix.os }}-${{ matrix.py-version }}-tf2.1.0-wheel
+          name: ${{ matrix.os }}-${{ matrix.py-version }}-tf2.2.0rc3-wheel
           path: ./dist
       - if: github.event_name != 'pull_request' || matrix.py-version == env.MIN_PY_VERSION  || matrix.py-version == env.MAX_PY_VERSION
         run: |

CONTRIBUTING.md

@@ -160,6 +160,8 @@ conda activate my_dev_environement
 Just run from the root:
 
 ```
+pip install tensorflow==2.2.0rc3
+# you can use "pip install tensorflow-cpu==2.2.0rc3" too if you're not testing on gpu.
 pip install -e ./
 ```
 
@@ -247,7 +249,7 @@ If you need a custom C++/Cuda op for your test, compile your ops with
 
 ```bash
 python configure.py
-pip install tensorflow==2.1.0 -e ./ -r tools/install_deps/pytest.txt
+pip install tensorflow==2.2.0rc3 -e ./ -r tools/install_deps/pytest.txt
 bash tools/install_so_files.sh  # Linux/macos/WSL2
 sh tools/install_so_files.sh    # PowerShell
 ```
@@ -275,14 +277,14 @@ docker run --runtime=nvidia --rm -it -v ${PWD}:/addons -w /addons tensorflow/ten
 
 Configure:
 ```
-python3 -m pip install tensorflow==2.1.0
+python3 -m pip install tensorflow==2.2.0rc3
 python3 ./configure.py  # Links project with TensorFlow dependency
 ```
 
 Install in editable mode
 ```
 python3 -m pip install -e .
-python3 -m pip install pytest pytest-xdist
+python3 -m pip install -r tools/install_deps/pytest.txt
 ```
 
 Compile the custom ops
@@ -309,9 +311,9 @@ quickly, as Bazel has great support for caching and distributed testing.
 To test with Bazel:
 
 ```
-python3 -m pip install tensorflow==2.1.0
+python3 -m pip install tensorflow==2.2.0rc3
 python3 configure.py
-python3 -m pip install pytest
+python3 -m pip install -r tools/install_deps/pytest.txt
 bazel test -c opt -k \
 --test_timeout 300,450,1200,3600 \
 --test_output=all \

README.md

@@ -71,8 +71,8 @@ what it was tested against.
 #### Python-Op Compatibility Matrix
 | TFA Version    | TensorFlow | Python  |
 |:----------------------- |:---|:---------- |
-| tfa-nightly | 2.1, 2.2 | 3.5, 3.6, 3.7, 3.8 | 
-| tensorflow-addons-0.9.1 | 2.1 |3.5, 3.6, 3.7, 3.8 |
+| tfa-nightly | 2.2 | 3.5, 3.6, 3.7, 3.8 | 
+| tensorflow-addons-0.9.1 | 2.1,2.2 |3.5, 3.6, 3.7, 3.8 |
 | tensorflow-addons-0.8.3 | 2.1 |3.5, 3.6, 3.7 |
 | tensorflow-addons-0.7.1 | 2.1 | 2.7, 3.5, 3.6, 3.7 | 
 | tensorflow-addons-0.6.0 | 2.0 | 2.7, 3.5, 3.6, 3.7 |
@@ -102,7 +102,7 @@ compiled differently. A typical reason for this would be conda installed TensorF
 #### Custom-Op Compatibility Matrix
 | TFA Version    | TensorFlow | Compiler  | cuDNN | CUDA | 
 |:----------------------- |:---- |:---------|:---------|:---------|
-| tfa-nightly | 2.1 | GCC 7.3.1 | 7.6 | 10.1 |
+| tfa-nightly | 2.2 | GCC 7.3.1 | 7.6 | 10.1 |
 | tensorflow-addons-0.9.1 | 2.1  | GCC 7.3.1 | 7.6 | 10.1 |
 | tensorflow-addons-0.8.3 | 2.1  | GCC 7.3.1 | 7.6 | 10.1 |
 | tensorflow-addons-0.7.1 | 2.1  | GCC 7.3.1 | 7.6 | 10.1 |

tensorflow_addons/utils/ensure_tf_install.py

@@ -23,7 +23,7 @@
 
 import tensorflow as tf
 
-MIN_TF_VERSION = "2.1.0"
+MIN_TF_VERSION = "2.2.0"
 MAX_TF_VERSION = "2.3.0"
 
 

tensorflow_addons/utils/resource_loader.py

@@ -20,8 +20,8 @@
 
 import tensorflow as tf
 
-MIN_TF_VERSION_FOR_ABI_COMPATIBILITY = "2.1.0"
-MAX_TF_VERSION_FOR_ABI_COMPATIBILITY = "2.2.0"
+MIN_TF_VERSION_FOR_ABI_COMPATIBILITY = "2.2.0"
+MAX_TF_VERSION_FOR_ABI_COMPATIBILITY = "2.3.0"
 abi_warning_already_raised = False
 SKIP_CUSTOM_OPS = False
 

tools/docker/cpu_tests.Dockerfile

@@ -1,7 +1,7 @@
 #syntax=docker/dockerfile:1.1.5-experimental
 FROM python:3.5 as build_wheel
 
-ARG TF_VERSION=2.1.0
+ARG TF_VERSION=2.2.0rc3
 RUN pip install 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.1.0
+tensorflow-cpu~=2.2.0rc3

tools/install_deps/tensorflow.txt

@@ -1 +1 @@
-tensorflow~=2.1.0
+tensorflow~=2.2.0rc3

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.1.0 \
+       --build-arg TF_VERSION=2.2.0rc3 \
        --build-arg PY_VERSION=3.5 \
        -t tfa_gpu_tests ./
 docker run --rm -t -v cache_bazel:/root/.cache/bazel --gpus=all tfa_gpu_tests