forked from 170010011/fr
25 lines
782 B
Python
25 lines
782 B
Python
|
|
import numpy
|
|
import os
|
|
import platform
|
|
|
|
|
|
def configuration(parent_package='', top_path=None):
|
|
from numpy.distutils.misc_util import Configuration
|
|
config = Configuration('datasets', parent_package, top_path)
|
|
config.add_data_dir('data')
|
|
config.add_data_dir('descr')
|
|
config.add_data_dir('images')
|
|
config.add_data_dir(os.path.join('tests', 'data'))
|
|
if platform.python_implementation() != 'PyPy':
|
|
config.add_extension('_svmlight_format_fast',
|
|
sources=['_svmlight_format_fast.pyx'],
|
|
include_dirs=[numpy.get_include()])
|
|
config.add_subpackage('tests')
|
|
return config
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from numpy.distutils.core import setup
|
|
setup(**configuration(top_path='').todict())
|