fix package name bug for nightly (#155)

Co-authored-by: dhuangnm <[email protected]>
vllm-project · Sep 10, 2024 · 051650d · 051650d
1 parent 108afb0
commit 051650d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -12,18 +12,18 @@
 )
 version_info = extract_version_info(package_path)
 
-if version_info.is_release:
+if version_info.build_type == "release":
     package_name = "llmcompressor"
-elif version_info.is_dev:
+elif version_info.build_type == "dev":
     package_name = "llmcompressor-dev"
-elif version_info.is_nightly:
+elif version_info.build_type == "nightly":
     package_name = "llmcompressor-nightly"
 else:
     raise ValueError(f"Unsupported build type {version_info.build_type}")
 
 
 setup(
-    name="llmcompressor",
+    name=package_name,
     version=version_info.version,
     author="Neuralmagic, Inc.",
     author_email="[email protected]",