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 os
import subprocess
from packaging import version
from filecmp import cmp
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'
RAW_TEXT_FILE_PREFIX = 'https://raw.githubusercontent.com/hpcaitech/public_assets/main/colossalai/torch_build/torch_wheels'
CUDA_HOME = os.environ['CUDA_HOME']