From 3bb9b02b5a7bbc0d43131dad2bbcf7c1e1e3984b Mon Sep 17 00:00:00 2001 From: tfeda Date: Fri, 31 Dec 2021 19:12:07 -0500 Subject: [PATCH 1/2] Add __version__ attribute to python package --- python/datafusion/__init__.py | 3 +++ python/datafusion/tests/test_imports.py | 2 +- 2 files changed, 4 insertions(+), 1 deletion(-) 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..d2153bc35d2bc 100644 --- a/python/datafusion/tests/test_imports.py +++ b/python/datafusion/tests/test_imports.py @@ -30,7 +30,7 @@ def test_import_datafusion(): assert datafusion.__name__ == "datafusion" - + assert datafusion.__version__ is not None def test_class_module_is_datafusion(): for klass in [ From 01dab7bcda8e4c203929c660e731846ed09ac318 Mon Sep 17 00:00:00 2001 From: tfeda Date: Sat, 1 Jan 2022 08:55:15 -0500 Subject: [PATCH 2/2] fix flake8 linting issues --- python/datafusion/tests/test_imports.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/datafusion/tests/test_imports.py b/python/datafusion/tests/test_imports.py index d2153bc35d2bc..d6b15d7f3724c 100644 --- a/python/datafusion/tests/test_imports.py +++ b/python/datafusion/tests/test_imports.py @@ -32,6 +32,7 @@ def test_import_datafusion(): assert datafusion.__name__ == "datafusion" assert datafusion.__version__ is not None + def test_class_module_is_datafusion(): for klass in [ ExecutionContext,