Added TF 2.6 support.

sebastian-sz · Aug 13, 2021 · 1e138dc · 1e138dc
1 parent 4378ae6
commit 1e138dc
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -45,7 +45,7 @@ jobs:
       runs-on: ubuntu-latest
       strategy:
         matrix:
-          tensorflow-version: [ 2.2, 2.3, 2.4, 2.5 ]
+          tensorflow-version: [ 2.2, 2.3, 2.4, 2.5, 2.6 ]
       steps:
         - uses: actions/checkout@v2
         - name: Set up Python 3.6

diff --git a/setup.py b/setup.py
@@ -18,7 +18,7 @@ def _package_exists(name: str) -> bool:
 def _get_tensorflow_requirement():
     """Avoid re-download and misdetection of package."""
     lower = 2.2
-    upper = 2.6
+    upper = 2.7
 
     if _package_exists("tensorflow-cpu"):
         return [f"tensorflow-cpu>={lower},<{upper}"]

diff --git a/tests/requirements.txt b/tests/requirements.txt
@@ -1,6 +1,6 @@
 # Onnx conversion
-tf2onnx~=1.9.1
-onnxruntime~=1.8.1
+tf2onnx>=1.9.1,<1.10
+onnxruntime>=1.8.1,<1.10
 
 # Tensorflow Model Optimization Toolkit
 tensorflow-model-optimization~=0.6.0