ran black

nerfstudio-project · Aug 21, 2024 · e227643 · e227643
1 parent 7fd8585
commit e227643
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gsplat/cuda/_torch_impl.py b/gsplat/cuda/_torch_impl.py
@@ -137,7 +137,9 @@ def _ortho_proj(
     J = torch.stack([fx, O, O, O, fy, O], dim=-1).reshape(C, 1, 2, 3).repeat(1, N, 1, 1)
 
     cov2d = torch.einsum("...ij,...jk,...kl->...il", J, covars, J.transpose(-1, -2))
-    means2d = means[..., :2] * Ks[:, None, [0, 1], [0, 1]] + Ks[:, None, [0, 1], [2, 2]] # [C, N, 2]
+    means2d = (
+        means[..., :2] * Ks[:, None, [0, 1], [0, 1]] + Ks[:, None, [0, 1], [2, 2]]
+    )  # [C, N, 2]
     return means2d, cov2d  # [C, N, 2], [C, N, 2, 2]