Fix typing error for blend_background

This commit is contained in:
Jonas Kulhanek committed 2023-07-17 14:14:02 +02:00
1 parent 220ef3a459
commit 9229b1f23b
1 file changed
+1 -1
+1 -1
View File
@@ -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).