File size: 1,305 Bytes
74b1bac |
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 38 39 40 |
from setuptools import setup, find_packages
package_name = "bm25s"
version = {}
with open(f"{package_name}/version.py", encoding="utf8") as fp:
exec(fp.read(), version)
with open("README.md", encoding="utf8") as fp:
long_description = fp.read()
extras_require = {
"core": ["jax[cpu]", "ujson", "tqdm", "PyStemmer"],
"stem": ["PyStemmer"],
"hf": ["huggingface_hub"],
"dev": ["black"],
}
# Dynamically create the 'full' extra by combining all other extras
extras_require["full"] = sum(extras_require.values(), [])
setup(
name=package_name,
version=version["__version__"],
author="Xing Han Lù",
author_email=f"{package_name}@googlegroups.com",
url=f"https://github.com/xhluca/{package_name}",
description=f"An ultra-fast implementation of BM25 based on sparse matrices.",
long_description=long_description,
packages=find_packages(include=[f"{package_name}*"]),
package_data={},
install_requires=['scipy', 'numpy'],
extras_require=extras_require,
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
],
python_requires=">=3.8",
# Cast long description to markdown
long_description_content_type="text/markdown",
) |