Retrieve the version number at runtime

dottxt-ai · Oct 9, 2024 · b755e4d · b755e4d
1 parent cf63eb5
commit b755e4d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -113,6 +113,7 @@ ignore_missing_imports = true
 [tool.coverage.run]
 omit = [
     "python/outlines_core/_version.py",
+    "python/outlines_core/__init__.py",
     "tests/*",
 ]
 branch = true

diff --git a/python/outlines_core/__init__.py b/python/outlines_core/__init__.py
@@ -1,4 +1,12 @@
 """Outlines is a Generative Model Programming Framework."""
+from importlib.metadata import PackageNotFoundError, version
+
 import outlines_core.models
 
+try:
+    __version__ = version("outlines_core")
+except PackageNotFoundError:
+    pass
+
+
 __all__ = ["models"]