From 34bd1e984dfa43998ba473204b8c8aa44a12ef8c Mon Sep 17 00:00:00 2001
From: chenfeiyu <chenfeiyu@baidu.com>
Date: Fri, 22 Nov 2019 11:32:59 +0800
Subject: [PATCH] add setup.py

---
 parakeet/__init__.py                    |  1 +
 parakeet/datasets/__init__.py           |  0
 parakeet/{data => datasets}/ljspeech.py |  8 ++---
 setup.py                                | 46 +++++++++++++++++++++++++
 tests/test_ljspeech.py                  |  2 +-
 5 files changed, 52 insertions(+), 5 deletions(-)
 create mode 100644 parakeet/datasets/__init__.py
 rename parakeet/{data => datasets}/ljspeech.py (93%)
 create mode 100644 setup.py

diff --git a/parakeet/__init__.py b/parakeet/__init__.py
index e69de29..6c8e6b9 100644
--- a/parakeet/__init__.py
+++ b/parakeet/__init__.py
@@ -0,0 +1 @@
+__version__ = "0.0.0"
diff --git a/parakeet/datasets/__init__.py b/parakeet/datasets/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/parakeet/data/ljspeech.py b/parakeet/datasets/ljspeech.py
similarity index 93%
rename from parakeet/data/ljspeech.py
rename to parakeet/datasets/ljspeech.py
index 4ee51de..964b933 100644
--- a/parakeet/data/ljspeech.py
+++ b/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):
diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..c6e0148
--- /dev/null
+++ b/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)
\ No newline at end of file
diff --git a/tests/test_ljspeech.py b/tests/test_ljspeech.py
index 59061f8..5f83722 100644
--- a/tests/test_ljspeech.py
+++ b/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