From 7310aede97a0fdacc0be3219b6f8174b53351075 Mon Sep 17 00:00:00 2001 From: Yineng Zhang Date: Wed, 11 Dec 2024 06:48:45 +0800 Subject: [PATCH] fix: compatible with PEP 440 (#2435) --- sgl-kernel/setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sgl-kernel/setup.py b/sgl-kernel/setup.py index 7672eb094..cc2375315 100644 --- a/sgl-kernel/setup.py +++ b/sgl-kernel/setup.py @@ -31,13 +31,13 @@ def rename_wheel(): zip_ref.extractall(tmp_dir) old_info = tmp_dir / f"sgl_kernel-{base_version}.dist-info" - new_info = tmp_dir / f"sgl_kernel-{base_version}+cu{cuda_version}.dist-info" + new_info = tmp_dir / f"sgl_kernel-{base_version}.post0+cu{cuda_version}.dist-info" old_info.rename(new_info) platform = "manylinux2014_x86_64" new_wheel = wheel_dir / old_wheel.name.replace("linux_x86_64", platform) new_wheel = wheel_dir / new_wheel.name.replace( - base_version, f"{base_version}+cu{cuda_version}" + base_version, f"{base_version}.post0+cu{cuda_version}" ) with zipfile.ZipFile(new_wheel, "w", zipfile.ZIP_DEFLATED) as new_zip: