fixed missing return type in simple_trainer.py (#126)

nerfstudio-project · Feb 13, 2024 · aa1ff65 · aa1ff65
1 parent 2d33520
commit aa1ff65
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/simple_trainer.py b/examples/simple_trainer.py
@@ -89,7 +89,7 @@ def train(self, iterations: int = 1000, lr: float = 0.01, save_imgs: bool = Fals
         times = [0] * 3  # project, rasterize, backward
         for iter in range(iterations):
             start = time.time()
-            xys, depths, radii, conics, num_tiles_hit, cov3d = _ProjectGaussians.apply(
+            xys, depths, radii, conics, compensation, num_tiles_hit, cov3d = _ProjectGaussians.apply(
                 self.means,
                 self.scales,
                 1,