Merge pull request #1484 from Jackwaterveg/r0.1_develop

[Setup] make librosa version fixed

paddlespeech/__init__.py

@@ -11,5 +11,3 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
-__version__ = '0.1.0'

setup.py

@@ -27,6 +27,8 @@
 
 HERE = Path(os.path.abspath(os.path.dirname(__file__)))
 
+VERSION = '0.1.2'
+
 requirements = {
     "install": [
         "editdistance",
@@ -37,7 +39,7 @@
         "jieba",
         "jsonlines",
         "kaldiio",
-        "librosa",
+        "librosa==0.8.1",
         "loguru",
         "matplotlib",
         "nara_wpe",
@@ -82,6 +84,24 @@
 }
 
 
+def write_version_py(filename='paddlespeech/__init__.py'):
+    import paddlespeech
+    if hasattr(paddlespeech,
+               "__version__") and paddlespeech.__version__ == VERSION:
+        return
+    with open(filename, "a") as f:
+        f.write(f"\n__version__ = '{VERSION}'\n")
+
+
+def remove_version_py(filename='paddlespeech/__init__.py'):
+    with open(filename, "r") as f:
+        lines = f.readlines()
+    with open(filename, "w") as f:
+        for line in lines:
+            if "__version__" not in line:
+                f.write(line)
+
+
 @contextlib.contextmanager
 def pushd(new_dir):
     old_dir = os.getcwd()
@@ -169,10 +189,12 @@ def run(self):
         sys.exit()
 
 
+write_version_py()
+
 setup_info = dict(
     # Metadata
     name='paddlespeech',
-    version='0.1.1',
+    version=VERSION,
     author='PaddlePaddle Speech and Language Team',
     author_email='[email protected]',
     url='https://github.com/PaddlePaddle/PaddleSpeech',
@@ -235,3 +257,5 @@ def run(self):
     })
 
 setup(**setup_info)
+
+remove_version_py()