More permissive input annotations for get_render()

nerfstudio-project · Nov 21, 2024 · 5a96e1e · 5a96e1e
1 parent 4fe9d42
commit 5a96e1e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/viser/_viser.py b/src/viser/_viser.py
@@ -433,8 +433,8 @@ def get_render(
         height: int,
         width: int,
         *,
-        wxyz: tuple[float, float, float, float],
-        position: tuple[float, float, float],
+        wxyz: tuple[float, float, float, float] | np.ndarray,
+        position: tuple[float, float, float] | np.ndarray,
         fov: float,
         transport_format: Literal["png", "jpeg"] = "jpeg",
     ) -> np.ndarray: ...
@@ -453,8 +453,8 @@ def get_render(
         height: int,
         width: int,
         *,
-        wxyz: tuple[float, float, float, float] | None = None,
-        position: tuple[float, float, float] | None = None,
+        wxyz: tuple[float, float, float, float] | np.ndarray | None = None,
+        position: tuple[float, float, float] | np.ndarray | None = None,
         fov: float | None = None,
         transport_format: Literal["png", "jpeg"] = "jpeg",
     ) -> np.ndarray: