| from setuptools import setup, find_packages | |
| setup( | |
| name="tevatron", | |
| version="0.0.1", | |
| packages=find_packages("src"), | |
| package_dir={"": "src"}, | |
| url="https://github.com/texttron/tevatron", | |
| license="Apache 2.0", | |
| author="Luyu Gao", | |
| author_email="luyug@cs.cmu.edu", | |
| description="Tevatron: A toolkit for learning and running deep dense retrieval models.", | |
| python_requires=">=3.7", | |
| install_requires=["transformers>=4.10.0", "datasets>=1.1.3"], | |
| ) |