diff --git a/examples/python-udwf.py b/examples/python-udwf.py
index 32f8fadaa..7d39dc1b8 100644
--- a/examples/python-udwf.py
+++ b/examples/python-udwf.py
@@ -15,6 +15,8 @@
 # specific language governing permissions and limitations
 # under the License.
 
+from __future__ import annotations
+
 import datafusion
 import pyarrow as pa
 from datafusion import col, lit, udwf
diff --git a/pyproject.toml b/pyproject.toml
index f416e02a5..d16a18aa6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -66,7 +66,7 @@ features = ["substrait"]
 
 # Enable docstring linting using the google style guide
 [tool.ruff.lint]
-select = ["E4", "E7", "E9", "F", "D", "W", "I"]
+select = ["E4", "E7", "E9", "F", "FA", "D", "W", "I"]
 
 [tool.ruff.lint.pydocstyle]
 convention = "google"
diff --git a/python/datafusion/io.py b/python/datafusion/io.py
index 7f3b77efa..3b6264948 100644
--- a/python/datafusion/io.py
+++ b/python/datafusion/io.py
@@ -17,6 +17,8 @@
 
 """IO read functions using global context."""
 
+from __future__ import annotations
+
 import pathlib
 
 import pyarrow
diff --git a/python/tests/test_udaf.py b/python/tests/test_udaf.py
index e69c77d3c..97cf81f3c 100644
--- a/python/tests/test_udaf.py
+++ b/python/tests/test_udaf.py
@@ -15,6 +15,8 @@
 # specific language governing permissions and limitations
 # under the License.
 
+from __future__ import annotations
+
 from typing import List
 
 import pyarrow as pa
diff --git a/python/tests/test_udwf.py b/python/tests/test_udwf.py
index 0ffa04179..2fea34aa3 100644
--- a/python/tests/test_udwf.py
+++ b/python/tests/test_udwf.py
@@ -15,6 +15,8 @@
 # specific language governing permissions and limitations
 # under the License.
 
+from __future__ import annotations
+
 import pyarrow as pa
 import pytest
 from datafusion import SessionContext, column, lit, udwf