diff --git a/requirements.txt b/requirements.txt index 0740c65..c475fd9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -tensorflow==2.1.0 +tensorflow==2.11.1 tensorflow_privacy==0.2.2 sentencepiece==0.1.85 smart_open==1.10.0 diff --git a/setup.py b/setup.py index da0d778..c0aa0e0 100644 --- a/setup.py +++ b/setup.py @@ -27,7 +27,7 @@ 'numpy==1.18.3' ], extras_require={ - 'tf': ['tensorflow==2.1.0'], + 'tf': ['tensorflow==2.11.1'], 'py36': ['dataclasses==0.7'] } ) diff --git a/test-requirements.txt b/test-requirements.txt index 4704072..e2610de 100644 --- a/test-requirements.txt +++ b/test-requirements.txt @@ -1,4 +1,4 @@ -tensorflow==2.1.0 +tensorflow==2.11.1 flake8 pytest pytest-cov