diff --git a/python/datafusion/__init__.py b/python/datafusion/__init__.py index 0a25592f80ae2..15e71b1d33b93 100644 --- a/python/datafusion/__init__.py +++ b/python/datafusion/__init__.py @@ -19,6 +19,7 @@ from typing import List import pyarrow as pa +import importlib.metadata from ._internal import ( AggregateUDF, @@ -29,6 +30,8 @@ ) +__version__ = importlib.metadata.version("datafusion") + __all__ = [ "DataFrame", "ExecutionContext", diff --git a/python/datafusion/tests/test_imports.py b/python/datafusion/tests/test_imports.py index 423800248a5ce..d6b15d7f3724c 100644 --- a/python/datafusion/tests/test_imports.py +++ b/python/datafusion/tests/test_imports.py @@ -30,6 +30,7 @@ def test_import_datafusion(): assert datafusion.__name__ == "datafusion" + assert datafusion.__version__ is not None def test_class_module_is_datafusion():