From d3ed03d66772cc7b885764f5ca14523a5daecd04 Mon Sep 17 00:00:00 2001 From: dhuangnm Date: Tue, 10 Sep 2024 09:21:23 -0400 Subject: [PATCH] fix package name bug for nightly --- setup.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index 68e21faf9..9bffd60de 100644 --- 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="support@neuralmagic.com",