From 9229b1f23bf1860ca61e3f950d4317766438b1d9 Mon Sep 17 00:00:00 2001 From: Jonas Kulhanek Date: Mon, 17 Jul 2023 14:14:02 +0200 Subject: [PATCH] Fix typing error for blend_background --- nerfstudio/model_components/renderers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/model_components/renderers.py b/nerfstudio/model_components/renderers.py index 93dadd41f..a1c6c8512 100644 --- a/nerfstudio/model_components/renderers.py +++ b/nerfstudio/model_components/renderers.py @@ -143,7 +143,7 @@ class RGBRenderer(nn.Module): def blend_background( self, image: Tensor, - background_color: Optional[Float[Tensor, "3"]] = None, + background_color: Optional[BackgroundColor] = None, ) -> Float[Tensor, "*bs 3"]: """Blends the background color into the image if image is RGBA. Otherwise no blending is performed (we assume opacity of 1).