Torchscript fix

Signed-off-by: Eric Kerfoot <[email protected]>

Author: ericspod

monai/networks/nets/segresnet_ds.py

@@ -218,11 +218,10 @@ def _forward(self, x: torch.Tensor) -> list[torch.Tensor]:
         outputs = []
         x = self.conv_init(x)
 
-        level: nn.ModuleDict
-        for level in self.layers:  # type: ignore[assignment]
-            x = level["blocks"](x)
+        for level in self.layers:
+            x = level["blocks"](x)  # type: ignore
             outputs.append(x)
-            x = level["downsample"](x)
+            x = level["downsample"](x)  # type: ignore
 
         if self.head_module is not None:
             outputs = self.head_module(outputs)
@@ -407,14 +406,13 @@ def _forward(self, x: torch.Tensor) -> Union[None, torch.Tensor, list[torch.Tens
         outputs: list[torch.Tensor] = []
 
         i = 0
-        level: nn.ModuleDict
-        for level in self.up_layers:  # type: ignore
-            x = level["upsample"](x)
+        for level in self.up_layers:
+            x = level["upsample"](x)  # type: ignore
             x += x_down.pop(0)
-            x = level["blocks"](x)
+            x = level["blocks"](x)  # type: ignore
 
             if len(self.up_layers) - i <= self.dsdepth:
-                outputs.append(level["head"](x))
+                outputs.append(level["head"](x))  # type: ignore
             i = i + 1
 
         outputs.reverse()