Added nightly versioning..

hongyunnchen · Feb 26, 2020 · 840dcb8 · 840dcb8
1 parent 82bc302
commit 840dcb8
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/release/setup.py b/release/setup.py
@@ -19,6 +19,7 @@
 
 import sys
 
+from datetime import date
 from setuptools import Extension
 from setuptools import find_packages
 from setuptools import setup
@@ -37,6 +38,7 @@ def finalize_options(self):
 
 
 REQUIRED_PACKAGES = ['cirq >= 0.7.0', 'tensorflow >= 2.1.0']
+CUR_VERSION = '0.2.0'
 
 
 class BinaryDistribution(Distribution):
@@ -46,20 +48,21 @@ def has_ext_modules(self):
         return True
 
 
+nightly = False
 if '--nightly' in sys.argv:
     nightly = True
     sys.argv.remove('--nightly')
-else:
-    nightly = False
 
+project_name = 'tensorflow-quantum'
+build_version = CUR_VERSION
 if nightly:
     project_name = 'tfq-nightly'
-else:
-    project_name = 'tensorflow-quantum'
+    build_version = CUR_VERSION + '.dev' + str(datetime.date.today()).replace(
+        '-', '')
 
 setup(
     name=project_name,
-    version='0.2.0',
+    version=build_version,
     description=
     'TensorFlow Quantum is a library for hybrid quantum-classical machine learning.',
     author='Google Inc.',