setup.py 1.3 KB
Newer Older
C
chenfeiyu 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
import os 
import io 
import re
from setuptools import setup, find_packages

def read(*names, **kwargs):
    with io.open(
        os.path.join(os.path.dirname(__file__), *names),
        encoding=kwargs.get("encoding", "utf8")
    ) as fp:
        return fp.read()


def find_version(*file_paths):
    version_file = read(*file_paths)
    version_match = re.search(r"^__version__ = ['\"]([^'\"]*)['\"]",
                              version_file, re.M)
    if version_match:
        return version_match.group(1)
    raise RuntimeError("Unable to find version string.")

VERSION = find_version('parakeet', '__init__.py')
long_description = read('README.md')

setup_info = dict(
    # Metadata
    name='parakeet',
    version=VERSION,
    author='PaddleSL Team',
    author_email='',
    url='https://github.com/PaddlePaddle',
    description='Speech synthesis tools and models based on Paddlepaddle',
    long_description=long_description,
    license='Apache 2',

    install_requires=[
C
chenfeiyu 已提交
37 38
        'numpy', 'nltk', 'inflect', 'librosa', 'unidecode', 'numba', 
        'tqdm', 'matplotlib', 'tensorboardX', 'tensorboard', 'scipy',
39
        'ruamel.yaml', 'pandas', 'sox', 'paddlepaddle-gpu', 
C
chenfeiyu 已提交
40 41 42
    ],

    # Package info
C
chenfeiyu 已提交
43
    packages=find_packages(exclude=('tests', 'tests.*')),
C
chenfeiyu 已提交
44 45 46 47 48

    zip_safe=True,
)

setup(**setup_info)