get the python version from env (#4729)

5eae67cb · DavidChan · GitHub · 6dbf9998 · 5eae67cb
Unverified Commit 5eae67cb authored Mar 28, 2025 by DavidChan Committed by GitHub Mar 27, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 1 deletion

sgl-kernel/setup.py sgl-kernel/setup.py +7 -1

No files found.
--- a/sgl-kernel/setup.py
+++ b/sgl-kernel/setup.py
@@ -235,6 +235,12 @@ ext_modules = [
    ),
 ]

+
+def get_py_ver():
+    major, minor = sys.version_info[:2]
+    return f"cp{major}{minor}"
+
+
 setup(
    name="sgl-kernel",
    version=_get_version(),
@@ -245,5 +251,5 @@ setup(
        "build_ext": BuildExtension.with_options(use_ninja=True),
        "build_py": CustomBuildPy,
    },
-    options={"bdist_wheel": {"py_limited_api": "cp39"}},
+    options={"bdist_wheel": {"py_limited_api": get_py_ver()}},
 )