diff --git a/setup.py b/setup.py index e56e6358..b0e6318c 100644 --- a/setup.py +++ b/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', ], )