Specify supported tensorflow versions, fix: #1904

hankcs · Aug 22, 2024 · 32ed29b · 32ed29b
1 parent 022d0fb
commit 32ed29b
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/setup.py b/setup.py
@@ -14,10 +14,6 @@
 
 FASTTEXT = 'fasttext-wheel==0.9.2'
 sys_version_info = sys.version_info
-if sys_version_info >= (3, 10):
-    TENSORFLOW = ['tensorflow>=2.8.0']
-else:
-    TENSORFLOW = ['tensorflow==2.6.0', 'keras==2.6.0', 'protobuf<3.19']
 
 TOKENIZERS = []
 if (sys_version_info.major, sys_version_info.minor) == (3, 6) and sys.platform in {'darwin', 'win32'}:
@@ -30,7 +26,7 @@
         'perin-parser>=0.0.12',
     ],
     'fasttext': [FASTTEXT],
-    'tf': [FASTTEXT, *TENSORFLOW]
+    'tf': [FASTTEXT, 'tensorflow>=2.6.0,<2.14']
 }
 extras_require['full'] = list(set(sum(extras_require.values(), [])))