update package versions

CederGroupHub · Oct 7, 2023 · 218c5da · 218c5da
1 parent 5709972
commit 218c5da
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 8 deletions.
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -1,3 +1,3 @@
-include  README.md
+include README.md
 include materials_entity_recognition/data/*.json
 include materials_entity_recognition/models/*
diff --git a/requirement.txt b/requirement.txt
@@ -1,6 +1,7 @@
-tensorflow>=2.4.0
-tensorflow-addons>=0.12.1
-transformers>=4.4.2
+tensorflow==2.7.0
+tensorflow-addons==0.17.1
+protobuf==3.19.6
+transformers>=4.11.3
 torch
 spacy
 chemdataextractor

diff --git a/setup.py b/setup.py
@@ -6,19 +6,20 @@
 
 if __name__ == "__main__":
     setup(name='materials_entity_recognition',
-          version=3.2,
+          version=3.3,
           author="Tanjin He",
           author_email="[email protected]",
           license="MIT License",
           packages=find_packages(),
           include_package_data=True,
           install_requires=[
-              'tensorflow>=2.4.0',
-              'tensorflow-addons>=0.12.1',
+              'tensorflow==2.7.0',
+              'tensorflow-addons==0.17.1',
+              'protobuf==3.19.6',
               'spacy',
               'chemdataextractor',
               'numpy',
-              'transformers>=4.4.2',
+              'transformers>=4.11.3',
               'torch',
               'psutil',
           ],