From 500a4614bc34390d06dad0a2a88887cb6a5ce8b1 Mon Sep 17 00:00:00 2001
From: tfeda <trentonfeda@gmail.com>
Date: Mon, 3 Jan 2022 14:31:06 -0500
Subject: [PATCH] add __version__ attribute to datafusion object

---
 datafusion/__init__.py           | 4 ++++
 datafusion/tests/test_imports.py | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/datafusion/__init__.py b/datafusion/__init__.py
index 0a25592..8174380 100644
--- a/datafusion/__init__.py
+++ b/datafusion/__init__.py
@@ -18,6 +18,7 @@
 from abc import ABCMeta, abstractmethod
 from typing import List
 
+import importlib.metadata
 import pyarrow as pa
 
 from ._internal import (
@@ -29,6 +30,9 @@
 )
 
 
+__version__ = importlib.metadata.version("datafusion")
+
+
 __all__ = [
     "DataFrame",
     "ExecutionContext",
diff --git a/datafusion/tests/test_imports.py b/datafusion/tests/test_imports.py
index 4238002..2a3a21c 100644
--- a/datafusion/tests/test_imports.py
+++ b/datafusion/tests/test_imports.py
@@ -32,6 +32,10 @@ def test_import_datafusion():
     assert datafusion.__name__ == "datafusion"
 
 
+def test_datafusion_python_version():
+    assert datafusion.__version__ is not None
+
+
 def test_class_module_is_datafusion():
     for klass in [
         ExecutionContext,