from torch.nn import Module


class Encoder(Module):
    def __init__(self):
        super().__init__()

    def forward(self):
        pass


class Decoder(Module):
    def __init__(self, encoder):
        super().__init__()
        self.encoder = encoder()

    def forward(self):
        latent_space = self.encoder()
        pass