forked from mabel-dev/opteryx
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
45 lines (40 loc) · 1.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
import numpy as np
from Cython.Build import cythonize
from setuptools import Extension, find_packages, setup
with open("opteryx/version.py", "r") as v:
vers = v.read()
exec(vers) # nosec
with open("README.md", "r") as rm:
long_description = rm.read()
try:
with open("requirements.txt") as f:
required = f.read().splitlines()
except FileNotFoundError:
# this sometimes fails - so put them here, but this needs to be maintained manually
required = ["cython", "numpy", "orjson", "cityhash", "sqloxide", "pyarrow"]
extensions = [
Extension(
name="cjoin",
sources=["opteryx/third_party/pyarrow_ops/cjoin.pyx"],
include_dirs=[np.get_include()],
),
# Extension(
# name="cythonize",
# sources=["opteryx/third_party/accumulation_tree/accumulation_tree.pyx"],
# )
# "mabel/data/internals/group_by.py",
]
setup(
name="opteryx",
version=__version__,
description="Serverless SQL Engine",
long_description=long_description,
long_description_content_type="text/markdown",
maintainer="Joocer",
author="joocer",
author_email="[email protected]",
packages=find_packages(include=["opteryx", "opteryx.*"]),
url="https://github.com/mabel-dev/opteryx/",
install_requires=required,
ext_modules=cythonize(extensions),
)