setup: organize deps + remove requests

This commit is contained in:
mehrad 2022-03-07 11:32:21 -08:00
parent e758eca5d9
commit c3a2e643e9
No known key found for this signature in database
GPG Key ID: AAF81F778210AE42
1 changed files with 10 additions and 8 deletions

View File

@ -48,17 +48,19 @@ setuptools.setup(
install_requires=[ install_requires=[
'numpy>=1.14.5', 'numpy>=1.14.5',
'torch>=1.9.0,<1.11.0', 'torch>=1.9.0,<1.11.0',
'transformers==4.17.0',
'datasets==1.18.4',
'pathos==0.2.8',
# for sts:
'sentence-transformers==2.2.0',
# for alignment:
'num2words>=0.5.10',
'dateparser>=1.0.0',
# for metrics:
'rouge_score>=0.0.4', 'rouge_score>=0.0.4',
'sacrebleu>=1,<3', 'sacrebleu>=1,<3',
'bert-score~=0.3', 'bert-score~=0.3',
'requests~=2.22',
'num2words>=0.5.10',
'dateparser>=1.0.0',
'datasets==1.18.4',
'seqeval==1.2.2', 'seqeval==1.2.2',
'transformers==4.17.0',
'sentence-transformers==2.2.0',
'pathos==0.2.8',
# for kf: # for kf:
'kfserving>=0.5.0', 'kfserving>=0.5.0',
'ray==1.6.0', 'ray==1.6.0',
@ -68,7 +70,7 @@ setuptools.setup(
'scikit-learn>=0.23,<2.0', 'scikit-learn>=0.23,<2.0',
'dill~=0.3', 'dill~=0.3',
'xgboost~=1.3', 'xgboost~=1.3',
# dialogues # for e2e dialogues:
'dialogues==0.0.2', 'dialogues==0.0.2',
], ],
) )