fix: compatible with PEP 440 (#2435)

sgl-project · Dec 10, 2024 · 7310aed · 7310aed
1 parent 5de9a58
commit 7310aed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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: