* correct a typo to fix #373 (#383)

nerfstudio-project · Sep 3, 2024 · 9c6e591 · 9c6e591
1 parent 61be4a7
commit 9c6e591
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_basic.py b/tests/test_basic.py
@@ -239,11 +239,11 @@ def test_projection(test_data, fused: bool, calc_compensations: bool, ortho: boo
         )
 
     # backward
-    v_means2d = torch.randn_like(means2d) * radii[..., None]
-    v_depths = torch.randn_like(depths) * radii
-    v_conics = torch.randn_like(conics) * radii[..., None]
+    v_means2d = torch.randn_like(means2d) * valid[..., None]
+    v_depths = torch.randn_like(depths) * valid
+    v_conics = torch.randn_like(conics) * valid[..., None]
     if calc_compensations:
-        v_compensations = torch.randn_like(compensations) * radii
+        v_compensations = torch.randn_like(compensations) * valid
     v_viewmats, v_quats, v_scales, v_means = torch.autograd.grad(
         (means2d * v_means2d).sum()
         + (depths * v_depths).sum()