Add standard Python module __version__ attr (#3111)

kuzudb · Mar 22, 2024 · 3f817f2 · 3f817f2
1 parent f9e1b12
commit 3f817f2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tools/python_api/src_py/__init__.py b/tools/python_api/src_py/__init__.py
@@ -56,7 +56,7 @@
 
 
 def __getattr__(name: str) -> str | int:
-    if name == "version":
+    if name in ("version", "__version__"):
         return Database.get_version()
     elif name == "storage_version":
         return Database.get_storage_version()
@@ -75,6 +75,7 @@ def __getattr__(name: str) -> str | int:
     "PreparedStatement",
     "QueryResult",
     "Type",
+    "__version__",
     "storage_version",
     "version",
 ]
diff --git a/tools/python_api/test/test_version.py b/tools/python_api/test/test_version.py
@@ -3,3 +3,4 @@ def test_version() -> None:
 
     assert kuzu.version != ""
     assert kuzu.storage_version > 0
+    assert kuzu.version == kuzu.__version__