Add version export to python lib

earth-mover · Oct 8, 2024 · 2f9de3c · 2f9de3c
1 parent 67388ca
commit 2f9de3c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/icechunk-python/python/icechunk/__init__.py b/icechunk-python/python/icechunk/__init__.py
@@ -8,6 +8,7 @@
 from zarr.core.sync import SyncMixin
 
 from ._icechunk_python import (
+    __version__,
     PyIcechunkStore,
     S3Credentials,
     SnapshotMetadata,
@@ -21,6 +22,7 @@
 )
 
 __all__ = [
+    "__version__",
     "IcechunkStore",
     "StorageConfig",
     "S3Credentials",

diff --git a/icechunk-python/python/icechunk/_icechunk_python.pyi b/icechunk-python/python/icechunk/_icechunk_python.pyi
@@ -235,3 +235,5 @@ async def pyicechunk_store_open_existing(
 def pyicechunk_store_from_bytes(
     bytes: bytes, read_only: bool
 ) -> PyIcechunkStore: ...
+
+__version__: str
diff --git a/icechunk-python/src/lib.rs b/icechunk-python/src/lib.rs
@@ -755,6 +755,7 @@ impl PyIcechunkStore {
 /// The icechunk Python module implemented in Rust.
 #[pymodule]
 fn _icechunk_python(m: &Bound<'_, PyModule>) -> PyResult<()> {
+    m.add("__version__", env!("CARGO_PKG_VERSION"))?;
     m.add_class::<PyStorageConfig>()?;
     m.add_class::<PyIcechunkStore>()?;
     m.add_class::<PyS3Credentials>()?;