diff options
Diffstat (limited to 'pint/registry_helpers.py')
-rw-r--r-- | pint/registry_helpers.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/pint/registry_helpers.py b/pint/registry_helpers.py index 7eee694..6b2f0e0 100644 --- a/pint/registry_helpers.py +++ b/pint/registry_helpers.py @@ -13,7 +13,7 @@ from __future__ import annotations import functools from inspect import signature from itertools import zip_longest -from typing import TYPE_CHECKING, Callable, TypeVar, Any +from typing import TYPE_CHECKING, Callable, TypeVar, Any, Union, Optional from collections.abc import Iterable from ._typing import F @@ -186,8 +186,8 @@ def _apply_defaults(func, args, kwargs): def wraps( ureg: UnitRegistry, - ret: str | Unit | Iterable[str | Unit | None] | None, - args: str | Unit | Iterable[str | Unit | None] | None, + ret: Optional[Union[str, Unit, Iterable[Optional[Union[str, Unit]]]]], + args: Optional[Union[str, Unit, Iterable[Optional[Union[str, Unit]]]]], strict: bool = True, ) -> Callable[[Callable[..., Any]], Callable[..., Quantity]]: """Wraps a function to become pint-aware. @@ -301,7 +301,7 @@ def wraps( def check( - ureg: UnitRegistry, *args: str | UnitsContainer | Unit | None + ureg: UnitRegistry, *args: Optional[Union[str, UnitsContainer, Unit]] ) -> Callable[[F], F]: """Decorator to for quantity type checking for function inputs. |