dist ver

NiftyPET · Jan 5, 2021 · 9fedd4d · 9fedd4d
1 parent ec29506
commit 9fedd4d
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/niftypet/nipet/__init__.py b/niftypet/nipet/__init__.py
@@ -2,6 +2,16 @@
 """initialise the NiftyPET NIPET package"""
 __author__      = ("Pawel J. Markiewicz", "Casper O. da Costa-Luis")
 __copyright__   = "Copyright 2020"
+# version detector. Precedence: installed dist, git, 'UNKNOWN'
+try:
+    from ._dist_ver import __version__
+except ImportError:
+    try:
+        from setuptools_scm import get_version
+
+        __version__ = get_version(root="../..", relative_to=__file__)
+    except (ImportError, LookupError):
+        __version__ = "UNKNOWN"
 
 import logging
 import os

diff --git a/pyproject.toml b/pyproject.toml
@@ -2,3 +2,7 @@
 requires = ["setuptools>=42", "wheel", "setuptools_scm[toml]>=3.4",
             "ninst>=0.8.0", "numpy>=1.14", "miutil[cuda]>=0.4.0",
             "scikit-build>=0.11.0", "cmake>=3.18", "ninja"]
+
+[tool.setuptools_scm]
+write_to = "niftypet/nipet/_dist_ver.py"
+write_to_template = "__version__ = '{version}'\n"