forked from modin-project/modin
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsetup.py
58 lines (50 loc) · 2.1 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
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from setuptools import setup, find_packages
import setuptools.command.build_ext as _build_ext
import sys
import platform
import pip
pip_major = int(pip.__version__.split(".")[0])
if pip_major < 10:
# https://github.com/pypa/pip/issues/5240
from pip import main
else:
from pip._internal import main
python_version = platform.python_version()[:3]
ray_whl = "ray" # Fall back on pip install if no relevant wheel exists
ray_whl_prefix = "https://s3-us-west-2.amazonaws.com/ray-wheels/latest/"
if sys.platform.startswith('linux'):
if python_version == "2.7":
ray_whl = ray_whl_prefix + \
"ray-0.4.0-cp27-cp27mu-manylinux1_x86_64.whl"
elif python_version == "3.3":
ray_whl = ray_whl_prefix + "ray-0.4.0-cp33-cp33m-manylinux1_x86_64.whl"
elif python_version == "3.4":
ray_whl = ray_whl_prefix + "ray-0.4.0-cp34-cp34m-manylinux1_x86_64.whl"
elif python_version == "3.5":
ray_whl = ray_whl_prefix + "ray-0.4.0-cp35-cp35m-manylinux1_x86_64.whl"
elif python_version == "3.6":
ray_whl = ray_whl_prefix + "ray-0.4.0-cp36-cp36m-manylinux1_x86_64.whl"
elif sys.platform == "darwin":
if python_version == "2.7":
ray_whl = ray_whl_prefix + "ray-0.4.0-cp27-cp27m-macosx_10_6_intel.whl"
elif python_version == "3.4":
ray_whl = ray_whl_prefix + "ray-0.4.0-cp34-cp34m-macosx_10_6_intel.whl"
elif python_version == "3.5":
ray_whl = ray_whl_prefix + "ray-0.4.0-cp35-cp35m-macosx_10_6_intel.whl"
elif python_version == "3.6":
ray_whl = ray_whl_prefix + "ray-0.4.0-cp36-cp36m-macosx_10_6_intel.whl"
class build_ext(_build_ext.build_ext):
def run(self):
main(['install', ray_whl])
setup(
name="modin",
version="0.0.2",
description="Modin: Pandas on Ray - Make your pandas code run faster with "
"a single line of code change.",
packages=find_packages(),
cmdclass={"build_ext": build_ext},
url="https://github.com/modin-project/modin",
install_requires=["pandas==0.22"])