-
Notifications
You must be signed in to change notification settings - Fork 119
/
setup.py
36 lines (32 loc) · 977 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
# 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, <=1.2.1",
"pandas>=1.0.5, <=1.3.5",
"numpy>=1.18.5, <=1.21.6",
"scipy>=1.4.1, <=1.7.3",
"attrs>=19.3.0",
"flatten_dict>=0.3.0",
"tensorboardX>=2.1.0",
"gcsfs==0.6.2",
"google-auth>=2.15.0",
"fsspec <= 2023.1.0"
]
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,
author_email="[email protected]",
packages=find_packages(exclude=["tests"]),
package_data={"allrank": ["config.json"]},
entry_points={"console_scripts": ['allRank = allrank.main:run']},
zip_safe=False,
)