Support TensorFlow 2.6 (#873)

OpenNMT · Aug 30, 2021 · d36be18 · d36be18
1 parent 16cc45c
commit d36be18
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 deletions.
diff --git a/.github/workflows/ci.yml b/.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)
 

diff --git a/README.md b/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`:
 

diff --git a/docs/installation.md b/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.
 

diff --git a/opennmt/version.py b/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():

diff --git a/setup.py b/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": [