MLBEDSW-5752: Fix version string when installing from shallow clone
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Change-Id: I288decbc0affa7f9475cabbd16cd20005e15e2a2
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 @@
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",