added uncertainties to typing

varchasgopalaswamy · Aug 26, 2023 · e4c7492 · e4c7492
1 parent ed950ab
commit e4c7492
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pint/_typing.py b/pint/_typing.py
@@ -14,7 +14,7 @@
 
 HAS_NUMPY = False
 if TYPE_CHECKING:
-    from .compat import HAS_NUMPY
+    from .compat import HAS_NUMPY, HAS_AUTOUNCERTAINTIES
 
 if HAS_NUMPY:
     from .compat import np
@@ -25,8 +25,12 @@
     Scalar: TypeAlias = Union[float, int, Decimal, Fraction]
     Array: TypeAlias = Never
 
+if HAS_AUTOUNCERTAINTIES:
+    from auto_uncertainties import Uncertainty
+else:
+    Uncertainty: TypeAlias = Never
 # TODO: Change when Python 3.10 becomes minimal version.
-Magnitude = Union[Scalar, Array]
+Magnitude = Union[Scalar, Array, Uncertainty]
 
 UnitLike = Union[str, dict[str, Scalar], "UnitsContainer", "Unit"]