From 2445279a084163dfd4ff009a43df7dc185319bd5 Mon Sep 17 00:00:00 2001 From: Frank Lee Date: Tue, 10 Jan 2023 12:10:13 +0800 Subject: [PATCH] [setup] refactored setup.py for dependency graph (#2413) --- setup.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index 5128b80e8..b9cd9e5e4 100644 --- a/setup.py +++ b/setup.py @@ -144,13 +144,16 @@ if build_cuda_ext: print(f'===== Building Extension {name} =====') ext_modules.append(builder_cls().builder()) -if is_nightly: +# always put not nightly branch as the if branch +# otherwise github will treat colossalai-nightly as the project name +# and it will mess up with the dependency graph insights +if not is_nightly: + version = get_version() + package_name = 'colossalai' +else: # use date as the nightly version version = datetime.today().strftime('%Y.%m.%d') package_name = 'colossalai-nightly' -else: - version = get_version() - package_name = 'colossalai' setup(name=package_name, version=version,