mirror of
https://github.com/vale981/ray
synced 2025-03-09 04:46:38 -04:00
36 lines
915 B
Python
36 lines
915 B
Python
|
from __future__ import absolute_import
|
||
|
|
from __future__ import division
|
||
|
|
from __future__ import print_function
|
||
|
|
|
||
|
|
import os
|
||
|
|
from setuptools import setup
|
||
|
|
from Cython.Build import cythonize
|
||
|
|
|
||
|
|
import numpy
|
||
|
|
|
||
|
|
pkg_dir = "cython_examples"
|
||
|
|
modules = ["cython_simple.pyx", "masked_log.pyx"]
|
||
|
|
install_requires = ["cython", "numpy"]
|
||
|
|
include_dirs = [numpy.get_include()]
|
||
|
|
|
||
|
|
# TODO: Need scipy to run BrainIAK example, but don't want to add additional
|
||
|
|
# dependencies
|
||
|
|
try:
|
||
|
|
import scipy # noqa
|
||
|
|
modules.append("cython_blas.pyx")
|
||
|
|
install_requires.append("scipy")
|
||
|
|
except ImportError as e: # noqa
|
||
|
|
pass
|
||
|
|
|
||
|
|
modules = [os.path.join(pkg_dir, module) for module in modules]
|
||
|
|
|
||
|
|
setup(
|
||
|
|
name=pkg_dir,
|
||
|
|
version="0.0.1",
|
||
|
|
description="Cython examples for Ray",
|
||
|
|
packages=[pkg_dir],
|
||
|
|
ext_modules=cythonize(modules),
|
||
|
|
install_requires=install_requires,
|
||
|
|
include_dirs=include_dirs
|
||
|
|
)
|