From 3ac6386187eeff2f8e4c8a7a2d5256321a8f47af Mon Sep 17 00:00:00 2001
From: huangyuxin <hyxin2014@126.com>
Date: Tue, 18 Jan 2022 11:31:10 +0000
Subject: [PATCH] add remove_version_py

---
 setup.py       | 14 ++++++++++++++
 setup_audio.py |  9 +++++++++
 2 files changed, 23 insertions(+)

diff --git a/setup.py b/setup.py
index 5c6a9e49..bf591258 100644
--- a/setup.py
+++ b/setup.py
@@ -91,6 +91,14 @@ def write_version_py(filename='paddlespeech/__init__.py'):
     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):
@@ -247,3 +255,9 @@ setup_info = dict(
     })
 
 setup(**setup_info)
+
+
+remove_version_py()
+
+
+
diff --git a/setup_audio.py b/setup_audio.py
index d6e9e892..4842c177 100644
--- a/setup_audio.py
+++ b/setup_audio.py
@@ -24,6 +24,13 @@ def write_version_py(filename='paddleaudio/__init__.py'):
     with open(filename, "a") as f:
         f.write(f"\n__version__ = '{VERSION}'\n")
 
+def remove_version_py(filename='paddleaudio/__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)
 
 write_version_py()
 
@@ -50,3 +57,5 @@ setuptools.setup(
         'soundfile >= 0.9.0',
         'colorlog',
     ], )
+
+remove_version_py()
\ No newline at end of file
-- 
GitLab