from setuptools import setup, Distribution

class BinaryDistribution(Distribution):

def has_ext_modules(foo):

return True

packages=['paddle',

'paddle.proto',

'paddle.trainer',

'paddle.trainer_config_helpers',

'paddle.utils',

'paddle.v2',

'paddle.v2.dataset',

'paddle.v2.reader',

'paddle.v2.master',

'paddle.v2.plot',

'paddle.v2.framework',

'paddle.v2.framework.proto',

'py_paddle']

setup_requires=["requests",

"numpy>=1.12",

"protobuf==3.1",

"recordio",

"matplotlib",

"rarfile",

"scipy>=0.19.0",

"Pillow",

"nltk>=3.2.2"]

if '${CMAKE_SYSTEM_PROCESSOR}' not in ['arm', 'armv7-a', 'aarch64']:

setup_requires+=["opencv-python"]

setup(name='paddlepaddle',

version='${PADDLE_VERSION}',

description='Parallel Distributed Deep Learning',

install_requires=setup_requires,

packages=packages,

package_data={

'paddle.v2.master': ['libpaddle_master.so'],

'paddle.v2.framework': ['core.so'],

'py_paddle':['*.py','_swig_paddle.so']

},

package_dir={

'': '${CMAKE_CURRENT_SOURCE_DIR}',

# The paddle.v2.framework.proto will be generated while compiling.

# So that package points to other directory.

'paddle.v2.framework.proto': '${PADDLE_BINARY_DIR}/paddle/framework',

'py_paddle': '${PADDLE_SOURCE_DIR}/paddle/py_paddle'

},

scripts=['${PADDLE_BINARY_DIR}/paddle/scripts/paddle'],

distclass=BinaryDistribution,

data_files=[('/usr/local/opt/paddle/bin',

['${PADDLE_BINARY_DIR}/paddle/scripts/paddle_usage',

'${PADDLE_BINARY_DIR}/paddle/trainer/paddle_trainer',

'${PADDLE_BINARY_DIR}/paddle/trainer/paddle_merge_model',

'${PADDLE_BINARY_DIR}/paddle/pserver/paddle_pserver_main'])]

)