From 5ce2a86b5c82081f76df01733c54ab0608f2eb69 Mon Sep 17 00:00:00 2001 From: Isaac Yang Date: Thu, 12 May 2022 12:09:37 -0700 Subject: [PATCH] Blossom test (#523) * For blossom-test * Separate version and package name --- setup.py | 20 ++++++++++++++++++-- 1 file changed, 18 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 7298f6a2d3..1361cbff6e 100644 --- a/setup.py +++ b/setup.py @@ -14,6 +14,7 @@ import os import shutil +import datetime from setuptools import find_packages, setup @@ -28,11 +29,26 @@ os.remove(os.path.join(this_directory, "nvflare", "poc.zip")) shutil.make_archive(base_name="poc", format="zip", root_dir=os.path.join(this_directory, "nvflare"), base_dir="poc") shutil.move("poc.zip", os.path.join(this_directory, "nvflare", "poc.zip")) -package_name = "nvflare" + +versions = versioneer.get_versions() +if versions["error"]: + today = datetime.date.today().timetuple() + year = today[0] % 1000 + month = today[1] + day = today[2] + version = f"0.0.{year:02d}{month:02d}{day:02d}" +else: + version = versions["version"] + +release = os.environ.get("NVFL_RELEASE") +if release == "1": + package_name = "nvflare" +else: + package_name = "nvflare-nightly" setup( name=package_name, - version=versioneer.get_version(), + version=version, cmdclass=versioneer.get_cmdclass(), description="Federated Learning Application Runtime Environment", url="https://github.com/NVIDIA/NVFlare",