setup: organize deps + remove requests
This commit is contained in:
parent
e758eca5d9
commit
c3a2e643e9
18
setup.py
18
setup.py
|
@ -48,17 +48,19 @@ setuptools.setup(
|
|||
install_requires=[
|
||||
'numpy>=1.14.5',
|
||||
'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',
|
||||
'sacrebleu>=1,<3',
|
||||
'bert-score~=0.3',
|
||||
'requests~=2.22',
|
||||
'num2words>=0.5.10',
|
||||
'dateparser>=1.0.0',
|
||||
'datasets==1.18.4',
|
||||
'seqeval==1.2.2',
|
||||
'transformers==4.17.0',
|
||||
'sentence-transformers==2.2.0',
|
||||
'pathos==0.2.8',
|
||||
# for kf:
|
||||
'kfserving>=0.5.0',
|
||||
'ray==1.6.0',
|
||||
|
@ -68,7 +70,7 @@ setuptools.setup(
|
|||
'scikit-learn>=0.23,<2.0',
|
||||
'dill~=0.3',
|
||||
'xgboost~=1.3',
|
||||
# dialogues
|
||||
# for e2e dialogues:
|
||||
'dialogues==0.0.2',
|
||||
],
|
||||
)
|
||||
|
|
Loading…
Reference in New Issue