Skip to content

Commit 66edcb5

Browse files
author
Fabio Ferreira
committed
fix: small ruff issue
1 parent de2b6bd commit 66edcb5

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

monai/networks/nets/unet.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525

2626
__all__ = ["UNet", "Unet"]
2727

28+
2829
class _ActivationCheckpointWrapper(nn.Module):
2930
def __init__(self, module: nn.Module) -> None:
3031
super().__init__()
@@ -35,6 +36,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
3536
return cast(torch.Tensor, checkpoint(self.module, x, use_reentrant=False))
3637
return cast(torch.Tensor, self.module(x))
3738

39+
3840
class UNet(nn.Module):
3941
"""
4042
Enhanced version of UNet which has residual units implemented with the ResidualUnit class.

0 commit comments

Comments
 (0)