github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

pull/1865/head
yuxuan-lou 2022-11-09 16:54:09 +08:00 committed by GitHub
parent 653b0a620e
commit cc27adceb0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 4 deletions
Show Stats 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']