fix: Fix wrong replacement

CAREamics · Dec 17, 2024 · cdcc368 · cdcc368
1 parent 57be075
commit cdcc368
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/careamics/models/unet.py b/src/careamics/models/unet.py
@@ -102,7 +102,7 @@ def __init__(
                 )
             )
             encoder_blocks.append(self.pooling)
-        self.encoder_blocks = nn.Modulelist(encoder_blocks)
+        self.encoder_blocks = nn.ModuleList(encoder_blocks)
 
     def forward(self, x: torch.Tensor) -> list[torch.Tensor]:
         """
@@ -222,7 +222,7 @@ def __init__(
                 )
             )
 
-        self.decoder_blocks = nn.Modulelist(decoder_blocks)
+        self.decoder_blocks = nn.ModuleList(decoder_blocks)
 
     def forward(self, *features: torch.Tensor) -> torch.Tensor:
         """