Type fix

ericspod · ericspod · commit 66f1b3a4866c · 2025-09-21T13:09:17.000+01:00
Signed-off-by: Eric Kerfoot &lt;17726042+ericspod@users.noreply.github.com&gt;
diff --git a/monai/networks/nets/diffusion_model_unet.py b/monai/networks/nets/diffusion_model_unet.py
@@ -2016,7 +2016,7 @@ def __init__(
 
         last_dim_flattened = int(reduce(lambda x, y: x * y, input_shape) * channels[-1])
 
-        self.out: Optional[nn.Module] = nn.Sequential(
+        self.out: nn.Module = nn.Sequential(
             nn.Linear(last_dim_flattened, 512), nn.ReLU(), nn.Dropout(0.1), nn.Linear(512, self.out_channels)
         )
 

Original file line number	Diff line number	Diff line change
`@@ -2016,7 +2016,7 @@ def __init__(`
`2016`	`2016`
`2017`	`2017`	`last_dim_flattened = int(reduce(lambda x, y: x * y, input_shape) * channels[-1])`
`2018`	`2018`
`2019`		`- self.out: Optional[nn.Module] = nn.Sequential(`
	`2019`	`+ self.out: nn.Module = nn.Sequential(`
`2020`	`2020`	`nn.Linear(last_dim_flattened, 512), nn.ReLU(), nn.Dropout(0.1), nn.Linear(512, self.out_channels)`
`2021`	`2021`	`)`
`2022`	`2022`