[CI] Fix setup.py (flashinfer-ai#101)

octoml · Jan 31, 2024 · 00aa360 · 00aa360
1 parent 0e5482f
commit 00aa360
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/setup.py b/python/setup.py
@@ -137,8 +137,10 @@ def get_instantiation_cu() -> list[str]:
 
 
 def get_version():
-    with open(root / "version.txt") as f:
-        version = f.read().strip()
+    version = os.getenv("FLASHINFER_BUILD_VERSION")
+    if version is None:
+        with open(root / "version.txt") as f:
+            version = f.read().strip()
     return version