forked from allegro/allRank
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
37 lines (33 loc) · 948 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# coding=utf-8
import pathlib
from setuptools import setup, find_packages
HERE = pathlib.Path(__file__).parent
README = (HERE / "README.md").read_text()
reqs = [
"scikit-learn>=1.1.0",
"pandas>=1.0.5",
"numpy>=1.18.5",
"scipy>=1.4.1",
"attrs>=19.3.0",
"flatten_dict>=0.3.0",
"tensorboardX>=2.1.0",
"gcsfs>=0.6.2",
"google-auth>=2.15.0",
"fsspec",
"torchvision",
]
setup(
name="allRank",
version="1.4.3",
description="allRank is a framework for training learning-to-rank neural models",
long_description=README,
long_description_content_type="text/markdown",
license="Apache 2",
url="https://github.com/allegro/allRank",
install_requires=reqs,
packages=find_packages(exclude=["tests"]),
package_data={"allrank": ["config.json"]},
entry_points={"console_scripts": ['allRank = allrank.main:run']},
zip_safe=False,
)