diff options
author | Rickard Bolin <rickard.bolin@arm.com> | 2021-12-16 17:03:59 +0000 |
---|---|---|
committer | Rickard Bolin <rickard.bolin@arm.com> | 2022-01-21 10:06:01 +0000 |
commit | 3c25ff658cf847ccfcb2d4d5796ffbe13a511894 (patch) | |
tree | 20805ef9d94dd36fec5fc2a7e62d839e6f28f362 /setup.py | |
parent | 36da8d32199bff60c493cd9ef7aa39a057c66304 (diff) | |
download | ethos-u-vela-3c25ff658cf847ccfcb2d4d5796ffbe13a511894.tar.gz |
MLBEDSW-5752: Fix version string when installing from shallow clone
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Change-Id: I288decbc0affa7f9475cabbd16cd20005e15e2a2
Diffstat (limited to 'setup.py')
-rw-r--r-- | setup.py | 8 |
1 files changed, 7 insertions, 1 deletions
@@ -18,6 +18,7 @@ import os import re +import setuptools_scm.git from setuptools import Extension from setuptools import setup from setuptools.command.build_ext import build_ext @@ -59,9 +60,14 @@ mlw_module = Extension( define_macros=[("NPY_NO_DEPRECATED_API", "NPY_1_9_API_VERSION")], ) + +def scm_version_parse(root): + return setuptools_scm.git.parse(root, pre_parse=setuptools_scm.git.fetch_on_shallow) + + setup( name="ethos-u-vela", - use_scm_version=True, + use_scm_version={"parse": scm_version_parse}, description="Neural network model compiler for Arm Ethos-U NPUs", long_description=long_description, long_description_content_type="text/markdown", |