diff --git a/src/affine/__init__.py b/src/affine/__init__.py
index 41c54ed..108f3a5 100644
--- a/src/affine/__init__.py
+++ b/src/affine/__init__.py
@@ -37,10 +37,20 @@
 from collections.abc import MutableSequence, Sequence
 from functools import cached_property
 import math
+from typing import TYPE_CHECKING, overload
 import warnings
 
 from attrs import astuple, define, field
 
+if TYPE_CHECKING:
+    from typing import TypeVar, overload
+
+    import numpy as np
+    from numpy.typing import NDArray
+
+    _ScalarType = TypeVar("_ScalarType", bound=np.generic)
+
+
 __all__ = ["Affine"]
 __author__ = "Sean Gillies"
 __version__ = "3.0rc2"
@@ -280,7 +290,7 @@ def permutation(cls, *scaling: float) -> Affine:
         """
         return cls(0.0, 1.0, 0.0, 1.0, 0.0, 0.0)
 
-    def __array__(self, dtype=None, copy: bool | None = None):
+    def __array__(self, dtype=None, copy: bool | None = None) -> NDArray[np.float64]:
         """Get affine matrix as a 3x3 NumPy array.
 
         Parameters