/
vshmidt
/
pytorch
Обзор
Документация
Войти
/
vshmidt
/
pytorch
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
.github/scripts/get_ci_variable.py
30 строк
829 B
atalman
Automate stable CUDA update and linter using min Python verison (#148912)
12 мар 2025, 21:12
12 мар 2025, 21:12
29fd875
Код
Авторство
О чём код?
#!/usr/bin/env python3 """Helper script - Return CI variables such as stable cuda, min python version, etc.""" import argparse import sys def main(args: list[str]) -> None: import generate_binary_build_matrix parser = argparse.ArgumentParser() parser.add_argument( "--cuda-stable-version", action="store_true", help="get cuda stable version", ) parser.add_argument( "--min-python-version", action="store_true", help="get min supported python version", ) options = parser.parse_args(args) if options.cuda_stable_version: return print(generate_binary_build_matrix.CUDA_STABLE) if options.min_python_version: return print(generate_binary_build_matrix.FULL_PYTHON_VERSIONS[0]) if __name__ == "__main__": main(sys.argv[1:])