|
14 | 14 | import errno |
15 | 15 | import io |
16 | 16 | import os |
| 17 | +import re |
17 | 18 | import subprocess |
18 | 19 | from datetime import datetime |
19 | 20 |
|
@@ -110,14 +111,30 @@ def show(): |
110 | 111 | f.write(content) |
111 | 112 |
|
112 | 113 |
|
113 | | -__version__ = "3.0.0b0.post" |
| 114 | +# only use this file to contral the version |
| 115 | +__version__ = "3.0.0b1.post" |
114 | 116 | if os.getenv(PADDLENLP_STABLE_VERSION): |
115 | 117 | __version__ = __version__.replace(".post", "") |
116 | 118 | else: |
117 | 119 | formatted_date = datetime.now().date().strftime("%Y%m%d") |
118 | 120 | __version__ = __version__.replace(".post", ".post{}".format(formatted_date)) |
119 | 121 |
|
120 | 122 |
|
| 123 | +# write the version information for the develop version |
| 124 | +def append_version_py(filename="paddlenlp/__init__.py"): |
| 125 | + assert os.path.exists(filename), f"{filename} does not exist!" |
| 126 | + |
| 127 | + with open(filename, "r") as file: |
| 128 | + file_content = file.read() |
| 129 | + |
| 130 | + pattern = r"^# \[VERSION_INFO\].*$" |
| 131 | + modified_content = re.sub(pattern, f'\n__version__ = "{__version__}"\n\n', file_content, flags=re.MULTILINE) |
| 132 | + with open(filename, "w") as file: |
| 133 | + file.write(modified_content) |
| 134 | + |
| 135 | + |
| 136 | +append_version_py(filename="paddlenlp/__init__.py") |
| 137 | + |
121 | 138 | extras = {} |
122 | 139 | REQUIRED_PACKAGES = read_requirements_file("requirements.txt") |
123 | 140 | extras["tests"] = read_requirements_file("tests/requirements.txt") |
@@ -197,6 +214,8 @@ def get_package_data_files(package, data, package_dir=None): |
197 | 214 | ) |
198 | 215 | except Exception as e: |
199 | 216 | git_checkout(paddlenlp_dir, "paddlenlp/version/__init__.py") if commit != "unknown" else None |
| 217 | + git_checkout(paddlenlp_dir, "paddlenlp/__init__.py") if commit != "unknown" else None |
200 | 218 | raise e |
201 | 219 |
|
202 | 220 | git_checkout(paddlenlp_dir, "paddlenlp/version/__init__.py") if commit != "unknown" else None |
| 221 | +git_checkout(paddlenlp_dir, "paddlenlp/__init__.py") if commit != "unknown" else None |
0 commit comments