From 51e0a6f22eeae337e291764e8b4c9c43782d5d47 Mon Sep 17 00:00:00 2001 From: Brent Yi Date: Sun, 5 May 2024 23:48:42 -0700 Subject: [PATCH] Fix type --- src/viser/transforms/utils/_utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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], ...]", )