Skip to content

Commit

Permalink
Blossom test (#523)
Browse files Browse the repository at this point in the history
* For blossom-test

* Separate version and package name
  • Loading branch information
IsaacYangSLA committed May 12, 2022
1 parent 192d180 commit 5ce2a86
Showing 1 changed file with 18 additions and 2 deletions.
20 changes: 18 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@

import os
import shutil
import datetime

from setuptools import find_packages, setup

Expand All @@ -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",
Expand Down

0 comments on commit 5ce2a86

Please sign in to comment.