From 679e2458055096a28c1fbcdbce397de20329e25f Mon Sep 17 00:00:00 2001 From: Jeremy Leibs Date: Tue, 4 Apr 2023 12:21:07 -0400 Subject: [PATCH] Allow torch tensors for log_rigid3 (#1769) --- 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]: