From 3c25ff658cf847ccfcb2d4d5796ffbe13a511894 Mon Sep 17 00:00:00 2001 From: Rickard Bolin Date: Thu, 16 Dec 2021 17:03:59 +0000 Subject: MLBEDSW-5752: Fix version string when installing from shallow clone Signed-off-by: Rickard Bolin Change-Id: I288decbc0affa7f9475cabbd16cd20005e15e2a2 --- setup.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 68fd28e..11af0b8 100644 --- a/setup.py +++ b/setup.py @@ -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", -- cgit v1.2.1