From b2ba7d44f5e8939e1fa878fe1d0ccf1af2cd3cef Mon Sep 17 00:00:00 2001 From: Jeremy Leibs Date: Tue, 4 Apr 2023 18:12:00 +0200 Subject: [PATCH] Allow torch tensors for log_rigid3 --- rerun_py/rerun_sdk/rerun/log/__init__.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/rerun_py/rerun_sdk/rerun/log/__init__.py b/rerun_py/rerun_sdk/rerun/log/__init__.py index b36abdacd5c6..fd7a74d1cf76 100644 --- a/rerun_py/rerun_sdk/rerun/log/__init__.py +++ b/rerun_py/rerun_sdk/rerun/log/__init__.py @@ -35,10 +35,7 @@ def _to_sequence(array: Optional[npt.ArrayLike]) -> Optional[Sequence[float]]: - if isinstance(array, np.ndarray): - return np.require(array, float).tolist() # type: ignore[no-any-return] - - return array # type: ignore[return-value] + return np.require(array, float).tolist() # type: ignore[no-any-return] def _normalize_colors(colors: Optional[Union[Color, Colors]] = None) -> npt.NDArray[np.uint8]: