Commit 190c3dd1 authored by Dr.李's avatar Dr.李

update setup.py script to include lib files for linux

parent 5876c8b3
...@@ -7,6 +7,7 @@ Created on 2017-4-25 ...@@ -7,6 +7,7 @@ Created on 2017-4-25
import platform import platform
import io import io
import os
from setuptools import setup from setuptools import setup
from setuptools import find_packages from setuptools import find_packages
from Cython.Build import cythonize from Cython.Build import cythonize
...@@ -35,6 +36,12 @@ if platform.system() != "Windows": ...@@ -35,6 +36,12 @@ if platform.system() != "Windows":
library_dirs=['alphamind/pfopt/lib'], library_dirs=['alphamind/pfopt/lib'],
extra_compile_args=['-std=c++11']), extra_compile_args=['-std=c++11']),
] ]
lib_files = []
lib_folder = 'alphamind/pfopt/lib'
for file in os.listdir(lib_folder):
lib_files.append(os.path.join(lib_folder, file))
else: else:
extensions = [ extensions = [
Extension('alphamind.cython.optimizers', Extension('alphamind.cython.optimizers',
...@@ -49,6 +56,14 @@ else: ...@@ -49,6 +56,14 @@ else:
extra_compile_args=['/MD']), extra_compile_args=['/MD']),
] ]
lib_files = []
lib_files = []
lib_folder = 'alphamind/pfopt/lib'
for file in os.listdir(lib_folder):
lib_files.append(os.path.join(lib_folder, file))
setup( setup(
name='Alpha-Mind', name='Alpha-Mind',
version=VERSION, version=VERSION,
...@@ -61,5 +76,7 @@ setup( ...@@ -61,5 +76,7 @@ setup(
install_requires=io.open('requirements.txt', encoding='utf8').read(), install_requires=io.open('requirements.txt', encoding='utf8').read(),
include_dirs=[np.get_include()], include_dirs=[np.get_include()],
ext_modules=cythonize(extensions), ext_modules=cythonize(extensions),
description='' data_files=[('alphamind/pfopt/lib', lib_files)],
description='',
include_package_data=True
) )
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment