Auto set python package version from git tag

JDAI-CV · May 23, 2019 · 51b19b2 · 51b19b2
1 parent bf69edf
commit 51b19b2
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tools/onnx2daq/python/setup.py b/tools/onnx2daq/python/setup.py
@@ -40,6 +40,13 @@
 tests_require = []
 extras_require = {}
 
+build_source_branch = os.getenv("BUILD_SOURCEBRANCH", "")
+if build_source_branch.startswith('refs/tags/v'):
+    dnnlibrary_version = build_source_branch[len('refs/tags/v'):]
+else:
+    dnnlibrary_version = '0.0.0'
+print("DNNLibrary version: {}".format(dnnlibrary_version))
+
 ################################################################################
 # Global variables for controlling the build variant
 ################################################################################
@@ -184,7 +191,7 @@ def build_extension(self, ext):
 
 setuptools.setup(
     name="onnx2daq",
-    # version=VersionInfo.version,
+    version=dnnlibrary_version,
     description="Convert ONNX to daq for DNNLibrary",
     ext_modules=ext_modules,
     cmdclass=cmdclass,