-
Notifications
You must be signed in to change notification settings - Fork 72
/
setup.py
87 lines (81 loc) · 2.33 KB
/
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
import os
import sys
from setuptools import find_packages, setup
from setuptools_rust import Binding, RustExtension
import versioneer
long_description = ""
if os.path.exists("README.md"):
with open("README.md") as f:
long_description = f.read()
needs_sphinx = "build_sphinx" in sys.argv
sphinx_requirements = ["sphinx>=3.2.1", "sphinx_rtd_theme"] if needs_sphinx else []
debug_build = "debug" in sys.argv
cmdclass = versioneer.get_cmdclass()
setup(
name="dask_sql",
version=versioneer.get_version(),
description="SQL query layer for Dask",
url="https://github.com/dask-contrib/dask-sql/",
maintainer="Nils Braun",
maintainer_email="nilslennartbraun@gmail.com",
license="MIT",
long_description=long_description,
long_description_content_type="text/markdown",
packages=find_packages(
include=["dask_sql", "dask_sql.*", "dask_planner", "dask_planner.*"]
),
package_data={"dask_sql": ["sql*.yaml"]},
rust_extensions=[
RustExtension(
"dask_planner.rust",
binding=Binding.PyO3,
path="dask_planner/Cargo.toml",
debug=debug_build,
)
],
python_requires=">=3.8",
setup_requires=sphinx_requirements,
install_requires=[
"dask[dataframe,distributed]>=2022.3.0,<=2022.9.1",
"pandas>=1.4.0",
"fastapi>=0.69.0",
"uvicorn>=0.11.3",
"tzlocal>=2.1",
"prompt_toolkit",
"pygments",
"tabulate",
"nest-asyncio",
],
extras_require={
"dev": [
"pytest>=6.0.1",
"pytest-cov>=2.10.1",
"mock>=4.0.3",
"sphinx>=3.2.1",
"pyarrow>=6.0.1",
"dask-ml>=2022.1.22",
"scikit-learn>=1.0.0",
"intake>=0.6.0",
"pre-commit",
"black==22.3.0",
"isort==5.7.0",
],
"fugue": ["fugue>=0.7.0"],
},
entry_points={
"console_scripts": [
"dask-sql-server = dask_sql.server.app:main",
"dask-sql = dask_sql.cmd:main",
],
"fugue.plugins": [
"dasksql = dask_sql.integrations.fugue:_register_engines[fugue]"
],
},
zip_safe=False,
cmdclass=cmdclass,
command_options={
"build_sphinx": {
"source_dir": ("setup.py", "docs"),
}
},
)