From 218c5da17cec0c2230dbd8140cbd198c110889cd Mon Sep 17 00:00:00 2001 From: Tanjin He Date: Fri, 6 Oct 2023 20:55:19 -0700 Subject: [PATCH] update package versions --- MANIFEST.in | 2 +- requirement.txt | 7 ++++--- setup.py | 9 +++++---- 3 files changed, 10 insertions(+), 8 deletions(-) diff --git a/MANIFEST.in b/MANIFEST.in index 83b16a0..1feb6c3 100644 --- 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/* \ No newline at end of file diff --git a/requirement.txt b/requirement.txt index 75b5f5d..cf7dc7d 100644 --- 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 index 01abfd6..efa3635 100644 --- 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="tanjin_he@berkeley.edu", 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', ],