mirror of
https://github.com/vale981/ray
synced 2025-03-08 11:31:40 -05:00
60d4d5e1aa
* Remove all __future__ imports from RLlib. * Remove (object) again from tf_run_builder.py::TFRunBuilder. * Fix 2xLINT warnings. * Fix broken appo_policy import (must be appo_tf_policy) * Remove future imports from all other ray files (not just RLlib). * Remove future imports from all other ray files (not just RLlib). * Remove future import blocks that contain `unicode_literals` as well. Revert appo_tf_policy.py to appo_policy.py (belongs to another PR). * Add two empty lines before Schedule class. * Put back __future__ imports into determine_tests_to_run.py. Fails otherwise on a py2/print related error.
30 lines
768 B
Python
30 lines
768 B
Python
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)
|