Add __version__ to the package (#53)

heavyai · Oct 12, 2021 · f0f5fc9 · f0f5fc9
1 parent 0508e47
commit f0f5fc9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/ibis_omniscidb/__init__.py b/ibis_omniscidb/__init__.py
@@ -2,6 +2,7 @@
 from __future__ import annotations
 
 import warnings
+from importlib.metadata import PackageNotFoundError, version
 from typing import Optional, Union
 
 import ibis.common.exceptions as com
@@ -24,12 +25,18 @@
 from .compiler import OmniSciDBCompiler
 from .udf import OmniSciDBUDF
 
+try:
+    __version__ = version("ibis_omniscidb")
+except PackageNotFoundError:
+    __version__ = ""
+
+
 try:
     from cudf import DataFrame as GPUDataFrame
 except (ImportError, OSError):
     GPUDataFrame = None
 
-__all__ = ('Backend',)
+__all__ = ('Backend', "__version__")
 
 
 class Backend(BaseSQLBackend):

diff --git a/ibis_omniscidb/tests/test_client.py b/ibis_omniscidb/tests/test_client.py
@@ -347,3 +347,7 @@ def test_current_database(con):
 def test_list_databases(con):
     assert 'ibis_testing' in con.list_databases()
     assert 'ibis_testing' in con.list_databases(like='ibis.*')
+
+
+def test_version():
+    assert ibis_omniscidb.__version__