From 339520c6e0ec4a2bd874250419d69eee225928e6 Mon Sep 17 00:00:00 2001 From: lucasliunju Date: Thu, 14 Jul 2022 10:41:01 +0800 Subject: [PATCH] [NFC] polish build_colossalai_wheel.py code style (#1306) --- .github/workflows/scripts/build_colossalai_wheel.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/.github/workflows/scripts/build_colossalai_wheel.py b/.github/workflows/scripts/build_colossalai_wheel.py index 7bd79fb64..dcedca737 100644 --- a/.github/workflows/scripts/build_colossalai_wheel.py +++ b/.github/workflows/scripts/build_colossalai_wheel.py @@ -62,7 +62,6 @@ def all_wheel_info(): wheel_info[torch_version][cuda_version][python_version] = dict(method=method, url=url, flags=flags) return wheel_info - def build_colossalai(wheel_info): cuda_version_major, cuda_version_minor = get_cuda_bare_metal_version() cuda_version_on_host = f'{cuda_version_major}.{cuda_version_minor}' @@ -114,4 +113,4 @@ if __name__ == '__main__': - \ No newline at end of file +