fix typing (#116)

nerfstudio-project · Feb 1, 2024 · 210ed53 · 210ed53
1 parent e13eff4
commit 210ed53
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gsplat/sh.py b/gsplat/sh.py
@@ -37,7 +37,7 @@ def spherical_harmonics(
     degrees_to_use: int,
     viewdirs: Float[Tensor, "*batch 3"],
     coeffs: Float[Tensor, "*batch D C"],
-) -> Float[Tensor, "*batch D C"]:
+) -> Float[Tensor, "*batch C"]:
     """Compute spherical harmonics
 
     Note: