diff --git a/mattport/nerf/renderers.py b/mattport/nerf/renderers.py index 887167d8b..081b2c4d2 100644 --- a/mattport/nerf/renderers.py +++ b/mattport/nerf/renderers.py @@ -19,8 +19,9 @@ class RendererOutputs: class RGBRenderer(nn.Module): """Standard volumetic rendering.""" + @classmethod def forward( - self, + cls, rgb: TensorType[..., "num_samples", 3], weights: TensorType[..., "num_samples"], ) -> RendererOutputs: