mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-04-29 12:15:39 +00:00
35 lines
862 B
Python
35 lines
862 B
Python
from datetime import datetime
|
|
|
|
|
|
def open_setup_file():
|
|
with open("setup.py", "r") as f:
|
|
file_lines = f.readlines()
|
|
return file_lines
|
|
|
|
|
|
def replace_nightly_package_info(file_lines):
|
|
version = datetime.today().strftime("%Y.%m.%d")
|
|
package_name = "colossalai-nightly"
|
|
|
|
for idx, line in enumerate(file_lines):
|
|
if "version = get_version()" in line:
|
|
file_lines[idx] = f'version = "{version}"\n'
|
|
if 'package_name = "colossalai"' in line:
|
|
file_lines[idx] = f'package_name = "{package_name}"\n'
|
|
return file_lines
|
|
|
|
|
|
def write_setup_file(file_lines):
|
|
with open("setup.py", "w") as f:
|
|
f.writelines(file_lines)
|
|
|
|
|
|
def main():
|
|
file_lines = open_setup_file()
|
|
file_lines = replace_nightly_package_info(file_lines)
|
|
write_setup_file(file_lines)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|