diff --git a/setup.py b/setup.py index 2917a1b..5e6e246 100644 --- a/setup.py +++ b/setup.py @@ -22,7 +22,7 @@ python_requires='>=3.6', install_requires=[ 'sentence-transformers', - 'pytrec_eval', + 'pytrec-eval-terrier', 'faiss_cpu', 'elasticsearch==7.9.1', 'datasets' @@ -36,4 +36,4 @@ "Topic :: Scientific/Engineering :: Artificial Intelligence" ], keywords="Information Retrieval Transformer Networks BERT PyTorch IR NLP deep learning" -) \ No newline at end of file +)