diff --git a/setup.py b/setup.py index 025d1825..8afcda85 100644 --- a/setup.py +++ b/setup.py @@ -48,7 +48,7 @@ setuptools.setup( install_requires=[ 'numpy>=1.14.5', 'torch>=1.9.0,<1.11.0', - 'pyrouge>=0.1.3', + 'rouge_score>=0.0.4', 'sacrebleu>=1,<3', 'bert-score~=0.3', 'requests~=2.22',