55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# Author: Taylor Smith <taylor.smith@alkaline-ml.com>
|
|
#
|
|
# Setup for submodules of pmdarima
|
|
|
|
import os
|
|
import sys
|
|
|
|
from pmdarima._build_utils import cythonize_extensions
|
|
|
|
|
|
# DEFINE CONFIG
|
|
def configuration(parent_package='', top_path=None):
|
|
from numpy.distutils.misc_util import Configuration
|
|
|
|
libs = []
|
|
if os.name == 'posix':
|
|
libs.append('m')
|
|
|
|
config = Configuration('pmdarima', parent_package, top_path)
|
|
|
|
# build utilities
|
|
config.add_subpackage('__check_build')
|
|
config.add_subpackage('_build_utils')
|
|
|
|
# submodules that do NOT have their own setup.py. manually add their tests
|
|
config.add_subpackage('compat')
|
|
config.add_subpackage('compat/tests')
|
|
config.add_subpackage('datasets')
|
|
config.add_subpackage('datasets/tests')
|
|
config.add_subpackage('model_selection')
|
|
config.add_subpackage('model_selection/tests')
|
|
|
|
# the following packages have cython or their own setup.py files.
|
|
config.add_subpackage('arima')
|
|
config.add_subpackage('preprocessing')
|
|
config.add_subpackage('utils')
|
|
|
|
# add test directory
|
|
config.add_subpackage('tests')
|
|
|
|
# Do cythonization, but only if this is not a release tarball, since the
|
|
# C/C++ files are not necessarily forward compatible with future versions
|
|
# of python.
|
|
if 'sdist' not in sys.argv:
|
|
cythonize_extensions(top_path, config)
|
|
|
|
return config
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from numpy.distutils.core import setup
|
|
setup(**configuration(top_path='').todict())
|