From 663a525d0ad3235c8690bfe9b6c6cd1afc2cb791 Mon Sep 17 00:00:00 2001 From: bdvllrs Date: Thu, 14 Dec 2023 10:13:45 +0000 Subject: [PATCH] Add latent_dim to abstract GWModule for auto-completion --- shimmer/modules/gw_module.py | 1 + 1 file changed, 1 insertion(+) diff --git a/shimmer/modules/gw_module.py b/shimmer/modules/gw_module.py index 25c2a760..e57dc67f 100644 --- a/shimmer/modules/gw_module.py +++ b/shimmer/modules/gw_module.py @@ -80,6 +80,7 @@ def forward(self, x: torch.Tensor) -> tuple[torch.Tensor, torch.Tensor]: class GWModule(nn.Module): domain_descr: Mapping[str, DomainDescription] + latent_dim: int def fusion_mechanism(self, x: Mapping[str, torch.Tensor]) -> torch.Tensor: raise NotImplementedError