| 10 | |
| 11 | |
| 12 | class Encoder(nn.Module, ABC, Generic[T]): |
| 13 | cfg: T |
| 14 | |
| 15 | def __init__(self, cfg: T) -> None: |
| 16 | super().__init__() |
| 17 | self.cfg = cfg |
| 18 | |
| 19 | @abstractmethod |
| 20 | def forward( |
| 21 | self, |
| 22 | context: BatchedViews, |
| 23 | ) -> Gaussians: |
| 24 | pass |
| 25 | |
| 26 | def get_data_shim(self) -> DataShim: |
| 27 | """The default shim doesn't modify the batch.""" |
| 28 | return lambda x: x |
nothing calls this directly
no outgoing calls
no test coverage detected