/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
.github/workflows/scripts/update_setup_for_nightly.py
34 строки
862 B
Frank Lee
[setup] fixed nightly release (#5388)
27 фев 2024, 10:19
Не верифицирован
27 фев 2024, 10:19
dcdd8a5
Код
Авторство
О чём код?
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()