MLBEDSW-5752: Fix version string when installing from shallow clone

Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Change-Id: I288decbc0affa7f9475cabbd16cd20005e15e2a2

1 files changed, 7 insertions, 1 deletions

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",