bugfix: Fix FLASHINFER_BUILD_VERSION in nightly build (#660)

flashinfer-ai · Dec 13, 2024 · 56f8eca · 56f8eca
1 parent 16549dc
commit 56f8eca
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -45,8 +45,11 @@
 
 
 def get_version():
+    build_version = os.environ.get("FLASHINFER_BUILD_VERSION")
+    if build_version is not None:
+        return build_version
     package_version = (root / "version.txt").read_text().strip()
-    local_version = os.getenv("FLASHINFER_LOCAL_VERSION")
+    local_version = os.environ.get("FLASHINFER_LOCAL_VERSION")
     if local_version is None:
         return package_version
     return f"{package_version}+{local_version}"