add setup.py

parakeet/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.0.0"

parakeet/datasets/ljspeech.py

@@ -4,10 +4,10 @@ import pandas as pd
 import librosa
 from .. import g2p
 
-from .sampler import SequentialSampler, RandomSampler, BatchSampler
-from .dataset import Dataset
-from .datacargo import DataCargo
-from .batch import TextIDBatcher, SpecBatcher
+from ..data.sampler import SequentialSampler, RandomSampler, BatchSampler
+from ..data.dataset import Dataset
+from ..data.datacargo import DataCargo
+from ..data.batch import TextIDBatcher, SpecBatcher
 
 
 class LJSpeech(Dataset):

setup.py

@@ -0,0 +1,46 @@
+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=[
+        'numpy', 'nltk', 'inflect', 'librosa', 'unidecode', 'numba', 'tqdm', 'matplotlib', 'tensorboardX', 'tensorboard', 'scipy', 'tqdm',
+    ],
+
+    # Package info
+    packages=find_packages(exclude=('test', 'test.*')),
+
+    zip_safe=True,
+)
+
+setup(**setup_info)

tests/test_ljspeech.py

@@ -1,4 +1,4 @@
-from parakeet.data.ljspeech import LJSpeech
+from parakeet.datasets.ljspeech import LJSpeech
 from parakeet.data.datacargo import DataCargo
 
 from pathlib import Path