github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-27 20:46:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NFC] polish .github/workflows/scripts/build_colossalai_wheel.py code style (#1856)

Browse Source
This commit is contained in:
yuxuan-lou
2022-11-09 16:54:09 +08:00
committed by GitHub
parent 653b0a620e
commit cc27adceb0
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
.github/workflows/scripts/build_colossalai_wheel.py vendored
View File

@@ -1,12 +1,13 @@
from filecmp import cmp
import requests
from bs4 import BeautifulSoup
import argparse import argparse
import os import os
import subprocess import subprocess
from packaging import version from filecmp import cmp
from functools import cmp_to_key from functools import cmp_to_key
import requests
from bs4 import BeautifulSoup
from packaging import version
WHEEL_TEXT_ROOT_URL = 'https://github.com/hpcaitech/public_assets/tree/main/colossalai/torch_build/torch_wheels' WHEEL_TEXT_ROOT_URL = 'https://github.com/hpcaitech/public_assets/tree/main/colossalai/torch_build/torch_wheels'
RAW_TEXT_FILE_PREFIX = 'https://raw.githubusercontent.com/hpcaitech/public_assets/main/colossalai/torch_build/torch_wheels' RAW_TEXT_FILE_PREFIX = 'https://raw.githubusercontent.com/hpcaitech/public_assets/main/colossalai/torch_build/torch_wheels'
CUDA_HOME = os.environ['CUDA_HOME'] CUDA_HOME = os.environ['CUDA_HOME']