Add curated_transformers.__version__ and use it for doc generation (#357

)
explosion · Dec 1, 2023 · 23f3a1b · 23f3a1b
1 parent 7a436e0
commit 23f3a1b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/curated_transformers/__init__.py b/curated_transformers/__init__.py
@@ -1,6 +1,14 @@
+import importlib.metadata
+
 import catalogue
 from catalogue import Registry
 
+__version__: str
+try:
+    __version__ = importlib.metadata.version("curated-transformers")
+except importlib.metadata.PackageNotFoundError:
+    __version__ = "unknown"
+
 
 class registry(object):
     """

diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -2,9 +2,13 @@
 
 # -- Project information
 
+import curated_transformers
+
 project = "Curated Transformers"
 copyright = "2021-2023, ExplosionAI GmbH"
 author = "ExplosionAI GmbH"
+version = curated_transformers.__version__
+release = curated_transformers.__version__
 
 # -- General configuration