From 51d63c91185a0b7609212d37b142e73ec5312825 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 2 Aug 2024 02:15:40 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Signed-off-by: am --- monai/networks/nets/cell_sam_wrapper.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/monai/networks/nets/cell_sam_wrapper.py b/monai/networks/nets/cell_sam_wrapper.py index 3da0e0b5e4..ccca07dee2 100644 --- a/monai/networks/nets/cell_sam_wrapper.py +++ b/monai/networks/nets/cell_sam_wrapper.py @@ -33,10 +33,10 @@ class CellSamWrapper(torch.nn.Module): auto_resize_inputs: whether to resize inputs before passing to the network. network_resize_roi: expected input size for the network. checkpoint: checkpoint file to load the SAM weights from. - return_features: whether to return features + return_features: whether to return features """ - + def __init__( self, auto_resize_inputs=True, @@ -77,7 +77,7 @@ def forward(self, x): if self.auto_resize_inputs: x = F.interpolate(x, size=self.network_resize_roi, mode="bilinear") - x = self.model.image_encoder(x) + x = self.model.image_encoder(x) if not self.return_features: x = self.model.mask_decoder(x)