Support TensorFlow 2.6 (#873)

Author: guillaumekln

.github/workflows/ci.yml

@@ -44,7 +44,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        tensorflow: [2.3, 2.4, 2.5]
+        tensorflow: [2.3, 2.4, 2.5, 2.6]
 
     steps:
       - uses: actions/checkout@v2
@@ -71,7 +71,7 @@ jobs:
           pytest --cov=opennmt --cov-report xml opennmt/tests
 
       - name: Upload coverage report
-        if: matrix.tensorflow == '2.5'
+        if: matrix.tensorflow == '2.6'
         run: |
           bash <(curl -s https://codecov.io/bash)
 

README.md

@@ -97,7 +97,7 @@ OpenNMT-tf also implements most of the techniques commonly used to train and eva
 OpenNMT-tf requires:
 
 * Python 3.5 or above
-* TensorFlow 2.3, 2.4, or 2.5
+* TensorFlow 2.3, 2.4, 2.5, or 2.6
 
 We recommend installing it with `pip`:
 

docs/installation.md

@@ -5,7 +5,7 @@
 OpenNMT-tf requires:
 
 * Python 3.5 or above
-* TensorFlow 2.3, 2.4, or 2.5
+* TensorFlow 2.3, 2.4, 2.5, or 2.6
 
 For GPU support, please read the [TensorFlow documentation](https://www.tensorflow.org/install/gpu) for additional software and hardware requirements.
 

opennmt/version.py

@@ -3,7 +3,7 @@
 __version__ = "2.20.1"
 
 INCLUSIVE_MIN_TF_VERSION = "2.3.0"
-EXCLUSIVE_MAX_TF_VERSION = "2.6.0"
+EXCLUSIVE_MAX_TF_VERSION = "2.7.0"
 
 
 def _check_tf_version():

setup.py

@@ -70,7 +70,7 @@ def get_project_version():
         "pyyaml>=5.3,<5.5",
         "rouge>=1.0,<2",
         "sacrebleu>=1.5.0,<2.1",
-        "tensorflow-addons>=0.13,<0.14",
+        "tensorflow-addons>=0.14,<0.15",
     ],
     extras_require={
         "tensorflow": [