Merge pull request #1832 from kadykov/transformation-typing

Fix Transformation typing
hgrecco · Aug 17, 2023 · 2852f36 · 2852f36
2 parents 6c2dda9 + c47e595
commit 2852f36
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/CHANGES b/CHANGES
@@ -5,7 +5,7 @@ Pint Changelog
 -----------------
 
 - Fixed Transformation type protocol.
-  (PR #1805)
+  (PR #1805, PR #1832)
 - Documented to_preferred and created added an autoautoconvert_to_preferred registry option.
   (PR #1803)
 - Optimize matplotlib unit conversion for Quantity arrays

diff --git a/pint/facets/context/objects.py b/pint/facets/context/objects.py
@@ -24,8 +24,8 @@
 
 class Transformation(Protocol):
     def __call__(
-        self, ureg: UnitRegistry, value: Magnitude, **kwargs: Any
-    ) -> Magnitude:
+        self, ureg: UnitRegistry, value: PlainQuantity, **kwargs: Any
+    ) -> PlainQuantity:
         ...