add version information to module

gautierdag · Dec 15, 2023 · 6ade316 · 6ade316
1 parent 4686982
commit 6ade316
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "bpeasy"
-version = "0.1.0"
+version = "0.1.1"
 edition = "2021"
 
 [lib]
@@ -17,6 +17,4 @@ fancy-regex = "0.12.0"
 pyo3 = { version = "0.19.0", features = ["extension-module"] }
 rayon = "1.8.0"
 regex = "1.5.4"
-serde_json = "1.0.108"
-
-
+serde_json = "1.0.108"
diff --git a/bpeasy/__init__.py b/bpeasy/__init__.py
@@ -1,8 +1,14 @@
+from importlib.metadata import version
+
 from .bpeasy import train_bpe
 
+__version__ = version("bpeasy")
+
+
 __all__ = [
     "save_vocab_to_tiktoken",
     "train_bpe",
+    "__version__",
 ]