fix lint errors

This commit is contained in:
evonneng committed 2022-04-09 05:31:52 +00:00
1 parent c7911281c3
commit 05d26b5265
2 files changed
+4 -2

No files matched your search

+1 -1
View File
@@ -1,2 +1,2 @@
from .encoding import Encoding
from .mlp import MLP
from .mlp import MLP
+3 -1
View File
@@ -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.