From 9a09f761bbdf7a2092fe998e7cd399bf6da6309d Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Thu, 21 Jan 2021 17:59:39 +0800 Subject: [PATCH] fix(whl): append buildid to wheel version GitOrigin-RevId: 00dabfa147e11e5f9a439da64a54de320eb9a238 --- imperative/python/setup.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/imperative/python/setup.py b/imperative/python/setup.py index eb97fe55..a90dfab0 100644 --- a/imperative/python/setup.py +++ b/imperative/python/setup.py @@ -11,6 +11,7 @@ import os import re import pathlib import platform +import subprocess from distutils.file_util import copy_file from setuptools import setup, find_packages, Extension from setuptools.command.build_ext import build_ext as _build_ext @@ -47,9 +48,11 @@ with open("megengine/version.py") as fp: __version__ = v['__version__'] email = 'megengine@megvii.com' +commit_id = subprocess.check_output(['git', 'rev-parse', '--short', 'HEAD']).strip().decode('ascii') +__version__ = '{}+1.git.{}'.format(__version__, commit_id) local_version = os.environ.get('LOCAL_VERSION') if local_version: - __version__ = '{}+{}'.format(__version__, local_version) + __version__ = '{}.{}'.format(__version__, local_version) packages = find_packages(exclude=['test']) megengine_data = [ -- GitLab