From 05d26b5265143f4d90568a30f7985543bfca87b7 Mon Sep 17 00:00:00 2001 From: evonneng Date: Sat, 9 Apr 2022 05:31:52 +0000 Subject: [PATCH] fix lint errors --- mattport/nerf/modules/__init__.py | 2 +- mattport/nerf/modules/encoding.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/mattport/nerf/modules/__init__.py b/mattport/nerf/modules/__init__.py index d59de93cb..39b25225c 100644 --- a/mattport/nerf/modules/__init__.py +++ b/mattport/nerf/modules/__init__.py @@ -1,2 +1,2 @@ from .encoding import Encoding -from .mlp import MLP \ No newline at end of file +from .mlp import MLP diff --git a/mattport/nerf/modules/encoding.py b/mattport/nerf/modules/encoding.py index 821c663c5..5abf0dbcf 100644 --- a/mattport/nerf/modules/encoding.py +++ b/mattport/nerf/modules/encoding.py @@ -10,7 +10,9 @@ class Encoding(nn.Module): def __init__(self, out_dim: int) -> None: super().__init__() - # raise NotImplementedError + self.out_dim = out_dim + + raise NotImplementedError def encode(self, in_tensor: TensorType[..., "input_dim"]) -> TensorType[..., "output_dim"]: """Encodes an input tensor.