diff --git a/imperative/python/setup.py b/imperative/python/setup.py index eb97fe557116008671f59f2e0610b94e9df25aac..a90dfab00a859ca31cfe94705b93e017434b708f 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 = [