diff --git a/src/viser/transforms/utils/_utils.py b/src/viser/transforms/utils/_utils.py index 765e51219..a25e0ac10 100644 --- a/src/viser/transforms/utils/_utils.py +++ b/src/viser/transforms/utils/_utils.py @@ -1,7 +1,6 @@ from typing import TYPE_CHECKING, Callable, Tuple, Type, TypeVar, Union, cast import numpy as onp -from jaxlie.hints import Array if TYPE_CHECKING: from .._base import MatrixLieGroup @@ -52,7 +51,7 @@ def _wrap(cls: Type[T]) -> Type[T]: TupleOfBroadcastable = TypeVar( "TupleOfBroadcastable", - bound="Tuple[Union[MatrixLieGroup, Array], ...]", + bound="Tuple[Union[MatrixLieGroup, onp.ndarray], ...]", )