From 96d749671b986089a385c4cce74c61ee9852ae3c Mon Sep 17 00:00:00 2001
From: Tomaz Vieira <tom.fogaca@gmail.com>
Date: Wed, 15 Jul 2026 10:45:26 +0000
Subject: [PATCH] Tightens type-checking for Transform

---
 src/transformnd/base.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/transformnd/base.py b/src/transformnd/base.py
index 666770b..e993d97 100644
--- a/src/transformnd/base.py
+++ b/src/transformnd/base.py
@@ -105,18 +105,18 @@ def apply(self, coords: ArrayT) -> ArrayT:
         """
         pass
 
-    def invert(self) -> Transform | None:
+    def invert(self) -> Transform[ArrayT] | None:
         """Invert the transformation, returning `None` if not possible."""
         return None
 
-    def __invert__(self) -> Transform:
+    def __invert__(self) -> Transform[ArrayT]:
         """Invert transformation if possible.
 
         Returns `NotImplemented` otherwise (will raise `NotImplementedError`).
 
         Returns
         -------
-        Transform
+        Transform[ArrayT]
             Inverted transformation.
         """
         t = self.invert()