26 lines
685 B
Python
26 lines
685 B
Python
import numpy
|
|
from numpy.distutils.misc_util import Configuration
|
|
from sklearn._build_utils import gen_from_templates
|
|
|
|
|
|
def configuration(parent_package="", top_path=None):
|
|
config = Configuration("_loss", parent_package, top_path)
|
|
|
|
# generate _loss.pyx from template
|
|
templates = ["sklearn/_loss/_loss.pyx.tp"]
|
|
gen_from_templates(templates)
|
|
|
|
config.add_extension(
|
|
"_loss",
|
|
sources=["_loss.pyx"],
|
|
include_dirs=[numpy.get_include()],
|
|
# define_macros=[("NPY_NO_DEPRECATED_API", "NPY_1_7_API_VERSION")],
|
|
)
|
|
return config
|
|
|
|
|
|
if __name__ == "__main__":
|
|
from numpy.distutils.core import setup
|
|
|
|
setup(**configuration().todict())
|